I'm trying CatBoost based on this article
In it's code, CatBoost has plot
in model.fit()
, so I want to try it on my IPython.
Here's my CatBoost code:
from catboost import CatBoostRegressor
# indicate categorical features for CatBoost
categorical_features_indices = np.where(X.dtypes != np.float)[0]
model=CatBoostRegressor(iterations=50, depth=3, learning_rate=0.1,
loss_function='RMSE')
model.fit(X_train, y_train,
cat_features=categorical_features_indices,
use_best_model=True,
eval_set=(X_validation, y_validation), plot=True)
But, it cannot show any plot and kept giving me error:
I did install ipywidgets and ipython. Do you know how to deal with this problem?