import lightgbm as lgb
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
# 导入数据
col_names = ["ID","K1K2驱动信号","电子锁驱动信号","急停信号","门禁信号","THDV-M","THDI-M","label"]
data = pd.read_csv("data_train.csv",names=col_names)
dataset_X = data[["K1K2驱动信号","电子锁驱动信号","急停信号","门禁信号","THDV-M","THDI-M"]].as_matrix()
dataset_Y = data[["label"]].as_matrix()
dataset_Y = np.array(dataset_Y).reshape(len(dataset_Y))
x_train, x_test, y_train, y_test = train_test_split(dataset_X, dataset_Y,test_size=0.2,)
# create dataset for lightgbm
lgb_train = lgb.Dataset(x_train, y_train)
lgb_eval = lgb.Dataset(x_test, y_test, reference=lgb_train)
param = {
'max_depth':8,
'num_leaves':16,
'learning_rate':0.4,
'scale_pos_weight':1,
'num_threads':8,
'objective':'binary',
'bagging_fraction':1,
'bagging_freq':1,
'min_sum_hessian_in_leaf':0.01
}
param['is_unbalance']='true'
param['metric'] = 'auc'
print('Start training...')
gbm = lgb.train(param,
lgb_train,
num_boost_round=500,
valid_sets=lgb_eval,
)
print('Start predicting...')
y_predict_test = gbm.predict(x_test)
print(y_predict_test)
def read_data():
col_names = ["ID", "K1K2驱动信号", "电子锁驱动信号", "急停信号", "门禁信号", "THDV-M", "THDI-M"]
data = pd.read_csv("data_test.csv", names=col_names)
# print(data.info())
data = data.fillna(0)
return data[["K1K2驱动信号", "电子锁驱动信号", "急停信号", "门禁信号", "THDV-M", "THDI-M"]],data["ID"]
subdata ,Id = read_data()
y_predict = gbm.predict(subdata)
print(y_predict)
for i in range(len(y_predict)):
if y_predict[i]>0.5:
y_predict[i]=1
else:
y_predict[i]=0
y_predict = [int(item) for item in y_predict]
print(y_predict)
submission = pd.DataFrame({
"id":Id,
"predictrion": y_predict
})
submission.to_csv("lgb_test1234.csv",index=None,header=None)
百度点石充电桩数据集+源码(acc=1)
需积分: 47 88 浏览量
2018-10-19
10:13:25
上传
评论 15
收藏 2.53MB RAR 举报
王小波_Libo
- 粉丝: 3322
- 资源: 14
最新资源
- stm32 usb接口通信
- Chessmate是一款完全免费的国际象棋学习软件,支持引擎分析,学开局、残局、棋书解读、大数据分析等功能
- 总结整理的Android面试Java基础知识点面试资料精编汇总文档资料合集.zip
- .android_lq
- FDN5632N-VB一款SOT23封装N-Channel场效应MOS管
- 毛老板-2404250902.amr
- Java类加载流程(双亲委派)流程图.zip
- FDN5632-NL-VB一款SOT23封装N-Channel场效应MOS管
- 新目标大学英语(第二版)视听说教程 第1册 Unit 4 TOP课件.zip
- 自动驾驶-状态估计和定位之Error State EKF.pdf
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈