Catboost calc_feature_statistics 未绘图

问题描述

在运行 catboost 网站 (https://catboost.ai/docs/concepts/python-reference_catboost_calc_feature_statistics.html) 上给出的示例脚本时,最后没有出现任何情节,只有空白。有什么建议可能是造成这种情况的原因吗?

from catboost import catboost
import numpy as np

train_data = np.random.rand(200,10)
label_values = np.random.rand(200)

model = catboost()
model.fit(train_data,label_values)

res = model.calc_feature_statistics(train_data,label_values,feature=2,plot=True)

解决方法

如果您不使用 jupyter,则默认情况下不会在您的控制台中呈现绘图。
就我而言,我使用 spyder 并偶然发现了同样的问题。
最简单的方法是将绘图保存到磁盘,而不是尝试在终端中显示它。

只需添加 plot_file 参数:

res = model.calc_feature_statistics(train_data,label_values,feature=2,plot_file="nameofyourplot.html)