如何解决如何使Python LightGBM代码接受列表
我正在使用以下代码:
import numpy as np
from sklearn.model_selection import train_test_split
import pandas as pd
from sklearn.preprocessing import StandardScaler
from sklearn.ensemble import GradientBoostingClassifier
from sklearn.metrics import mean_squared_error,roc_auc_score,precision_score
pd.options.display.max_columns = 999
import lightgbm as lgb
def load_csv(filepath):
data = []
col = []
checkcol = False
with open(filepath) as f:
for val in f.readlines():
val = val.replace("\n","")
val = val.split(',')
if checkcol is False:
col = val
checkcol = True
else:
data.append(val)
df = pd.DataFrame(data=data,columns=col)
return df
heart=load_csv(r'C:\Users\PC\Documents\Essay\heart.csv')
df=heart[['chol','cp']]
Y=heart['sex']
sc=StandardScaler()
sc.fit(df)
X=pd.DataFrame(sc.fit_transform(df))
X_train,X_test,y_train,y_test=train_test_split(X,Y,test_size=0.3,random_state=0)
d_train=lgb.Dataset(X_train,label=y_train)
params={}
params['learning_rate']=0.03
params['boosting_type']='gbdt' #GradientBoostingDecisionTree
params['objective']='binary' #Binary target feature
params['metric']='binary_logloss' #metric for binary classification
params['max_depth']=10
clf=lgb.train(params,d_train,100)
ValueError: Series.dtypes must be int,float or bool
我知道这是我选择Y的原因,但是我也尝试过使用数组和嵌套列表,但仍然失败。
解决方法
使用labelencoder可将您的列转换为期望的格式:
from sklearn import preprocessing
encoder = preprocessing.LabelEncoder()
encoder.fit(df['sex'])
encoder.transform(df['sex'])
这将产生一个零和一个清单,您可以将其输入到学习算法中:
array([1,1])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。