Skip to content

Commit

Permalink
Update __init__.py
Browse files Browse the repository at this point in the history
  • Loading branch information
SiddhantSadangi authored Apr 25, 2023
1 parent d6e4163 commit efac9e3
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/neptune_sklearn/impl/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,8 +110,8 @@ def create_regressor_summary(regressor, X_train, X_test, y_train, y_test, nrows=
log_charts (`bool`, optional): Whether to calculate and log chart visualizations.
Note: Calculating visualizations is potentially expensive depending on input data and regressor,
and may take some time to finish. This is equivalent to calling the following functions from
this module: `log_learning_curve_chart()`, `log_feature_importance_chart()`, `log_residuals_chart()`,
`log_prediction_error_chart()`, and `log_cooks_distance_chart()`.
this module: `create_learning_curve_chart()`, `create_feature_importance_chart()`, `create_residuals_chart()`,
`create_prediction_error_chart()`, and `create_cooks_distance_chart()`.
Returns:
`dict` with all summary items.
Expand Down Expand Up @@ -182,8 +182,8 @@ def create_classifier_summary(classifier, X_train, X_test, y_train, y_test, nrow
log_charts (`bool`, optional): Whether to calculate and log chart visualizations.
Note: Calculating visualizations is potentially expensive depending on input data and classifier, and
may take some time to finish. This is equivalent to calling the following functions from this module:
`log_classification_report_chart()`, `log_confusion_matrix_chart()`, `log_roc_auc_chart()`,
`log_precision_recall_chart()`, and `log_class_prediction_error_chart()`.
`create_classification_report_chart()`, `create_confusion_matrix_chart()`, `create_roc_auc_chart()`,
`create_precision_recall_chart()`, and `create_class_prediction_error_chart()`.
Returns:
`dict` with all summary items.
Expand Down

0 comments on commit efac9e3

Please sign in to comment.