lightgbm java_LightGBM两种使用方式

原生形式使用lightgbm(import lightgbm as lgb)

import lightgbm as lgb

from sklearn.metrics import mean_squared_error

from sklearn.datasets import load_iris

from sklearn.model_selection import train_test_split

# 加载数据

iris = load_iris()

data = iris.data

target = iris.target

# 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(data, target, test_size=0.2)

print("Train data length:", len(X_train))

print("Test data length:", len(X_test))

# 转换为Dataset数据格式

lgb_train = lgb.Dataset(X_train, y_train)

lgb_eval = lgb.Dataset(X_test, y_test, reference=lgb_train)

# 参数

params = {

'task': 'train',

'boosting_type': 'gbdt', # 设置提升类型

'objective': 'regression', # 目标函数

'met