我正在根据这篇文章尝试 CatBoost
在它的代码中,CatBoost 有plot
in model.fit()
,所以我想在我的 IPython 上尝试一下。
这是我的 CatBoost 代码:
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)
但是,它不能显示任何情节并不断给我错误:
我确实安装了 ipywidgets 和 ipython。你知道如何处理这个问题吗?