def test_model(self, model_file=None):
if model_file is None:
modelfile = self.model_file
else:
modelfile = self.single_model_path + 'model_' + str(model_file)
fault_test_file_path = get_test_data_path()
df = pd.read_csv(fault_test_file_path, encoding='utf-8', index_col=0)
data = df.iloc[:, :].values
x = data[:, :-1]
y = data[:, -1]
xgb = XGBRegressor()
raw_model = xgb.load_model(modelfile)
pred = raw_model.predict(x)
self.true = y
self.pred = pred
self.show_save_figure(fig_path=self.fault_data_test_figure_path,
modelname=self.job_name,
detal_idx=10)
t_mean = self.cal_mean(self.true)
p_mean = self.cal_mean(self.pred)
self.save_result(self.fault_data_test_result_path,
true_mean=t_mean,
pred_mean=p_mean)
