From 75718132254c5b7f1a544a0182ba5e2e6587ea4a Mon Sep 17 00:00:00 2001 From: Piotr Date: Tue, 8 Jun 2021 12:55:26 +0200 Subject: [PATCH] add info about predicted class order (#400) --- README.md | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/README.md b/README.md index 90c7b917..91928c13 100644 --- a/README.md +++ b/README.md @@ -355,6 +355,13 @@ In the above example the regression model will be fitted. + +
How to know the order of classes for binary or multiclass problem when using predict_proba? + +To get predicted probabilites with information about class label please use the `predict_all()` method. It returns the pandas DataFrame with class names in the columns. The order of predicted columns is the same in the `predict_proba()` and `predict_all()` methods. The `predict_all()` method will additionaly have the column with the predicted class label. + +
+ # Documentation For details please check [mljar-supervised docs](https://supervised.mljar.com).