找到lgbm超参数并训练模型

LGBtrainer的Python项目详细描述


lgbtrainer帮助您找到lgbm的超参数,并简化了训练模型和查找超参数的过程。

  • 参数:-
  1. train=应该是您的train数据集(适合培训目的)
  2. test=它应该是您的测试数据集(适合测试目的)
  3. y_train=它应该是您的目标列或值(与train相同的行)
  4. cv=分割或折叠的次数(用于查找超参数+训练模型)
  5. num_rounds=训练回合数(用于查找超参数+训练模型)
  6. metric=只能使用'auc'和'rmse'(目前只支持这两个)
  7. objective=可以提供“二进制”或“回归”或任何其他选项
  8. max_eval=为查找参数而执行的求值次数(注意:-根据数据集的大小,较大的数值可能需要更多的时间)
  • 示例:-
-from LGBtrainer import Model
-model = Model(train, test, y_train, metric='auc', objective='binary', max_eval=3, cv=5)
-params = model.get_params()
-predictions = model.lgb_model(params)

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java如何在表被注释到配置之前获取表的元数据?   java滚动条不会出现在JList上   java JOGL监视器GPU内存   java为什么要使用RecyclerView onDraw延迟   java定制Oppo Reno 2 Z CPH1951(手机型号)的固件(闪存文件)   java自定义线程池执行器   java如何解决发布版本中重复的jar条目[com/安卓/volley/R.class]?   java如何使用Bukkit API触发事件?   java在blazemeter jmeter RTE插件中使用ctrl+w输入   C#/Visual Studio的java JDT等价物   java为什么当maxread值很大而收到的消息数量很小时,卡夫卡消费者会无限期消费?   java游戏2。x:包含模板列表的绑定模型   带压缩的java日志旋转   运行时。exec用java运行程序读取它正在做什么