\begin{table}[!h]
\caption{Transfer Learning (off-diagonal) -- \textsc{H$^2$MN} (RMSE); \pki{diagonals show cases of train and test sets drawn from the same dataset.}}\label{tab:transferability_h2mn}
% \vspace{-2mm}
    \centering
    \scalebox{0.75}{
        \begin{tabular}{p{2.3cm}p{1cm}p{2.3cm}p{2.5cm}|p{1.3cm}}
        \toprule
        & \multicolumn{3}{c}{\textbf{Train Set}} & \textbf{\name}\\
        \textbf{Test Set} & \texttt{AIDS} & \texttt{ogbg-molhiv} & \texttt{ogbg-molpcba}\\
        \midrule
        \texttt{AIDS} & \textcolor{lightgray}{1.14} & 3.61 & 2.90 & 0.83\\
        \texttt{ogbg-molhiv} &  NA & \textcolor{lightgray}{12.01} & \cellcolor{yellow}{11.01} & 2.93\\
        \texttt{ogbg-molpcba} & NA & 5.95 & \textcolor{lightgray}{5.50} & 2.98\\
        \bottomrule
        \end{tabular}}
% \vspace{-4mm}
\end{table}
Test Set AIDS ogbg-molhiv ogbg-molpcba
AIDS 1.14 3.61 2.90
ogbg-molhiv NA 12.01 11.01
ogbg-molpcba NA 5.95 5.50 0.83
2.93
2.98