extend model building and prediction to accept feature_dataset_uri (instead of feature_generation_uri) again