Skip to content

Instantly share code, notes, and snippets.

@pycaret

pycaret/predict_model.py

Created August 1, 2020 01:53
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save pycaret/818cbf3945134873316a68456124a09b to your computer and use it in GitHub Desktop.
Save pycaret/818cbf3945134873316a68456124a09b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict_model.py
# 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