this should be "train" not "predict"