python的lightgbm模块里面好几个模型,一个是lightgbm.LGBMClassifier,还有一个是lightgbm.LGBMRegressor,此外还有一个是lightgbm.LGBMModel。我现在是要做分类,到底是用LGBMClassifier还是LGBMModel?这两个效果有什么区别吗?
1个回答
直接用的话,分类是LGBMClassifier,回归是LGBMRegressor。
如果要用LGBMModel,要在参数objective指定你的任务类型(regression, binary, multiclass, lambdarank),我一般都是直接用LGBMClassifier或者LGBMRegressor,没用过LGBMModel,不过我觉得指定了任务类型之后,其他参数一样的情况下,效果应该是一样的,因为LGBMClassifier是继承自LGBMModel,objective默认设置为binary或者multiclass。
SofaSofa数据科学社区DS面试题库 DS面经