Skip to content

Instantly share code, notes, and snippets.

@pycaret
Created August 1, 2020 01:53
Show Gist options
  • Save pycaret/818cbf3945134873316a68456124a09b to your computer and use it in GitHub Desktop.
Save pycaret/818cbf3945134873316a68456124a09b to your computer and use it in GitHub Desktop.
# train a catboost model
catboost = create_model('catboost')
# predict on holdout set (when no data is passed)
pred_holdout = predict_model(catboost)
# predict on new dataset
new_data = pd.read_csv('new-data.csv')
pred_new = predict_model(catboost, data = new_data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment