test_basic.py 682 Bytes
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
import numpy as np
from sklearn import datasets, metrics, model_selection
Guolin Ke's avatar
Guolin Ke committed
3
import lightgbm as lgb
Guolin Ke's avatar
Guolin Ke committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17


X, Y = datasets.make_classification(n_samples=100000, n_features=100)
x_train, x_test, y_train, y_test = model_selection.train_test_split(X, Y, test_size=0.1)

train_data = lgb.Dataset(x_train, max_bin=255, label=y_train)
valid_data = train_data.create_valid(x_test, label=y_test)

config={"objective":"binary","metric":"auc", "min_data":1, "num_leaves":15}
bst = lgb.Booster(params=config, train_set=train_data)
bst.add_valid(valid_data,"valid_1")

for i in range(100):
	bst.update()
Guolin Ke's avatar
Guolin Ke committed
18
19
20
	if i % 10 == 0:
		print(bst.eval_train())
		print(bst.eval_valid())
Guolin Ke's avatar
Guolin Ke committed
21
22
bst.save_model("model.txt")