predict_classification(ds: dataset, model: model_classification[ds], {
    "param": value,
    ...
}) -> (*predicted: column)

Usually employed after the train_classification step. ???+ info “Prediction Model” To use this step successfully you need to make sure the dataset you’re predicting on is as similar as possible to the one the model was trained on. We check that the necessary data types and columns are present, but you should pay attention to how you handled these in the recipe the model was generated. Any changes might lead to a significant degradation in model performance.

predict_classification(ds: dataset, model: model_classification[ds], {
    "param": value,
    ...
}) -> (*predicted: column)
​
positive_class
string

Name of the positive class. In binary classification, usually the class you’re most interested in, for example the label/class corresponding to successful lead conversion in a lead score model, the class corresponding to a customer who has churned in a churn prediction model, etc.

If provided, will return predicted probabilities for the positive class. If not provided, will return probabilities for the predicted class (i.e. the ‘winning’ class with the highest probability).

Was this page helpful?

Suggest edits