lightGBM预测相同的价值

Question

我对lgb有一个问题。当我写

def remove_na():
    df.dropna(subset=['Column 1', 'Column 2'])
    df.reset_index(drop=True)

df = remove_rows()
df.head(3)

AttributeError: 'NoneType' object has no attribute 'dropna'

它在不到一毫秒的时间内完成。 （对于（10000,25））形状数据集。

当我写预测时，所有输出变量都具有相同的值。

lgb.train(.......)

当我打印pred时，所有值都是（0.49）

这是我第一次使用 lightgbm 模块。我的代码有错误吗？或者我应该在数据集中寻找一些不匹配的地方。

Answer 1

您的 num_round 太小，它只是开始学习并在此停止。除此之外，请使您的verbose_eval较小，以便在训练时直观地看到结果。我建议您尝试如下的lgb.train代码：

clf = lgb.train（params，d_train，num_boost_round = 5000，verbose_eval = 10，early_stopping_rounds = 3500）

始终使用 early_stopping_rounds ，因为如果没有明显的学习经验或模型开始过拟合，则模型应停止。

请随时询问更多信息。玩得开心。