1. 程式人生 > >mac:jupyter notebook使用lightgbm

mac:jupyter notebook使用lightgbm

最近習慣了用jupyter notebook做比賽,但是使用lightgbm這種模組的時候,報錯說“找不到相關模組”,於是各種谷歌百度,最終解決了這個問題,下面和大家分享一下,也給自己做一下筆記。

1.lightgbm的安裝教程在前一篇的部落格有寫,這裡不在贅述。

 2.開啟你的jupyter notebook,輸入一下程式碼:

import lightgbm as lgb
from sklearn import datasets
from sklearn.model_selection import train_test_split
iris=datasets.load_iris()
X_train,X_test,y_train,y_test=train_test_split(iris.data,iris.target,test_size=0.3)
import numpy as np
train_data=lgb.Dataset(X_train,label=y_train)
validation_data=lgb.Dataset(X_test,label=y_test)
params={
    'learning_rate':0.1,
    'lambda_l1':0.1,
    'lambda_l2':0.2,
    'max_depth':4,
    'objective':'multiclass',
    'num_class':3,  #lightgbm.basic.LightGBMError: b‘Number of classes should be specified and greater than 1 for multiclass training‘
}
clf=lgb.train(params,train_data,valid_sets=[validation_data])
from sklearn.metrics import roc_auc_score,accuracy_score
y_pred=clf.predict(X_test)
y_pred=[list(x).index(max(x)) for x in y_pred]
print(y_pred)
print(accuracy_score(y_test,y_pred))

如果系統報錯:找不到相關模組。那就是你沒有配置好你的lightgbm

3.找到你在安裝好的 LightGBM 資料夾,拷貝里面的 lib_lightgbm.so 檔案到 LightGBM/python-package/lightgbm 目錄下。最後將 LightGBM/python-package/lightgbm 資料夾拷貝到你需要用到的 python 資料夾下。如果你不知道怎麼檢視python路徑,請開啟terminal,輸入一下程式碼:

input : python  --->import sys  ----> print sys.path

下面的輸出就是你的python檔案路徑。

4.像我的檔案路徑類似這樣:

/Users/aligege/anaconda3/lib/python3.6/site-packages

5.這個時候開啟您的資料夾,也可以使用快捷鍵command+shift+g直接進入此資料夾。找不到別心急,別忘了command+shift+.顯示隱藏檔案。

6.這個時候重啟一下jupyter notebook就ojbk啦。

參考:一位童鞋的分享