【文件属性】:
文件名称:lightgbm:Ruby的高性能梯度提升
文件大小:62KB
文件格式:ZIP
更新时间:2021-05-03 14:13:57
machine-learning lightgbm rubyml Ruby
轻型GBM
高性能渐变增强-适用于Ruby
安装
将此行添加到您的应用程序的Gemfile中:
gem 'lightgbm'
在Mac上,还要安装OpenMP:
brew install libomp
培训API
准备数据
x = [ [ 1 , 2 ] , [ 3 , 4 ] , [ 5 , 6 ] , [ 7 , 8 ] ]
y = [ 1 , 2 , 3 , 4 ]
训练模型
params = { objective : "regression" }
train_set = LightGBM :: Dataset . new ( x , label : y )
booster = LightGBM . train ( params , train_set )
预测
booster . predict ( x )
将模型保存到文件
booster . save_mode
【文件预览】:
lightgbm-master
----vendor()
--------LICENSE(1KB)
----Rakefile(2KB)
----.github()
--------workflows()
----test()
--------regressor_test.rb(2KB)
--------classifier_test.rb(4KB)
--------test_helper.rb(2KB)
--------support()
--------cv_test.rb(2KB)
--------ranker_test.rb(698B)
--------booster_test.rb(2KB)
--------dataset_test.rb(2KB)
--------train_test.rb(4KB)
----Gemfile(180B)
----.gitignore(103B)
----lightgbm.gemspec(557B)
----CHANGELOG.md(1KB)
----lib()
--------lightgbm()
--------lightgbm.rb(6KB)
----README.md(3KB)
----LICENSE.txt(1KB)