Fixed a bug in train_predict_2_options.