#?__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‘）
#?記錄程序運行時間
start_time?=?time.time（）

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

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

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

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

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