import joblib #引入可以调用模型的库
import pandas as pd

import xgboost as xgb

model = joblib.load('xgboost.pkl')

""" 输入path """

datapath = "data_template.csv"
result_path = "data_result1.csv"
dataset_use = pd.read_csv(datapath) #需要预测的测试集合的导入，更换文件就可以更换test_case了
dtest=xgb.DMatrix(dataset_use) # 输入成为xgbDMatrix
y_pred=model.predict(dtest) #输入模型进行验证预测
y_use = pd.DataFrame(y_pred)+1 #y数值的预测并且返还标签
y_use.columns = ['Label'] #
last_result = pd.concat([dataset_use,y_use],axis=1)
last_result.to_csv(result_path)