-
-
Notifications
You must be signed in to change notification settings - Fork 563
Description
Describe the bug
Can't plot class report with trained model
I have fitted model, y_test, y_pred, X_train=X_train, y_train=y_train, X_test= X_test, y_test=y_test.
run this code.
catboost_model = catboost_12 #trained in pycaret
model = wrap(catboost_model)
confusion_matrix(model,X_train=X_train, y_train=y_train, X_test= X_test, y_test=y_test, is_fitted=True)
TypeError: score() got an unexpected keyword argument 'sample_weight'
without wrapper it does not work too.
To Reproduce
# Steps to reproduce the behavior (code snippet):
# Should include imports, dataset loading, and execution
# Add the traceback below
Dataset
Did you use a specific dataset to produce the bug? Where can we access it?
Expected behavior
A clear and concise description of what you expected to happen.
Traceback
If applicable, add the traceback from the exception.
Desktop (please complete the following information):
- OS: [e.g. macOS]
- Python Version [e.g. 2.7, 3.6, miniconda]
- Yellowbrick Version [e.g. 0.7]
Additional context
Add any other context about the problem here.