#?__author__=‘chenyibin‘
#?-*-?coding:utf-8?-*-

import?sys
import?numpy?as?np
import?pandas?as?pd
import?xgboost?as?xgb
import?matplotlib.pyplot?as?plt
from?sklearn.metrics?import?mean_squared_error
import?time

reload（sys）
sys.setdefaultencoding（‘utf8‘）
#?記錄程序運(yùn)行時(shí)間
start_time?=?time.time（）

#?讀入數(shù)據(jù)
filename?=?‘齊礦數(shù)據(jù)集.csv‘
data_all?=?pd.read_csv（filename）

#?刪除相關(guān)度小的屬性（嘗試）
data_all?=?data_all.drop（“臺(tái)日效率“?1）
data_all?=?data_all.drop（“開(kāi)動(dòng)天數(shù)“?1）
data_all?=?data_all.drop（“日歷臺(tái)時(shí)“?1）
data_all?=?data_all.drop（“故障修“?1）
data_all?=?data_all.drop（“月計(jì)劃運(yùn)輸量“?1）

#?刪除缺失值
#?data_all.dropna（inplace=True）

#?從data_all中分離X和Y
X?=?data_all.drop（“單耗（千克/噸公里）“?1）
Y?=?data_all[“單耗（千克/噸公里）“]

#?劃分訓(xùn)練集、測(cè)試集
X_train?=?np.array（X[:800]）
X_test?=?np.array（X[800:]）
y_train?=?np.array（Y[:800]）
y_test?=?np.array（Y[800:]）