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啦。

参考:一位童鞋的分享

猜你喜欢

转载自blog.csdn.net/weixin_32087115/article/details/82757380