Added option to train only the hydrophobicity index