X_trin=X1[0:5000]
y_train=Y1[0:5000]
gbm = lgb.LGBMClassifier(boosting_type='gbdt',max_depth=-1,objective='multiclass',
num_leaves=5,learning_rate=0.6,n_estimators=1500,metric= 'multi_error',max_bin=150,verbose=-1)
gbm.fit(X_train, y_train,eval_set=[(X1, Y1)],early_stopping_rounds=500,eval_metric='multi_error')
y_pred = gbm.predict(X_train, pred_leaf=True)
print(y_pred.shape)上面的是LGB的参数，训练样本5000，标签四分类0,1,2,3
上面是运行结果 我想请教下1308代表的是什么 我看网上说y_pred[:,0] 输出的才是index 但是我不知道为什么
1个回答
你把关键代码复制一下，不然也不知道你的y_pred是什么意思
如果你要得到每个叶子的输出结果，你在调用模型进行predict或者predict_proba的时候设置一下参数，例如：
clf = LGBMClassifier(...)
leaves = clf.predict(x_test, pred_leaf=True)
黑泽先生
2020-03-20 08:43
你好，我把补充的发下面了，麻烦你看下，谢谢啦
-
classifier
2020-03-20 20:34
您的发帖不符合社区规范，问题补充可以直接对原问题修改。现在已帮您修复。谢谢配合。
-
SofaSofa驹炜
2020-03-20 21:42
相关讨论
随便看看