From 9c3e2718a5345d2de57f6fd8fc28956fddd1f50e Mon Sep 17 00:00:00 2001 From: Guolin Ke Date: Thu, 1 Dec 2016 19:14:22 +0800 Subject: [PATCH] Update simple_example.py --- examples/python-guide/simple_example.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/examples/python-guide/simple_example.py b/examples/python-guide/simple_example.py index ea7f99237..0ce96b057 100644 --- a/examples/python-guide/simple_example.py +++ b/examples/python-guide/simple_example.py @@ -8,3 +8,10 @@ rng = np.random.RandomState(2016) X, y = datasets.make_classification(n_samples=10000, n_features=100) x_train, x_test, y_train, y_test = model_selection.train_test_split(X, y, test_size=0.1, random_state=1) lgb_model = lgb.LGBMClassifier(n_estimators=100).fit(x_train, y_train, [(x_test, y_test)], eval_metric="auc") +lgb_model.predict(x_test) +# save model +lgb_model.booster().save_model('model.txt') +# load model +booster = lgb.Booster(model_file='model.txt') +# predict +print(booster.predict(x_test))