mac:jupyter notebook使用lightgbm
阿新 • • 發佈:2018-12-07
最近習慣了用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啦。
參考:一位童鞋的分享