def sub_st(): print('training ST...') RF_st1 = RF.RF_ST('./data/Normalized_Data/testing_DataSetST.csv', './data/Normalized_Data/sub_DataSetST.csv') RF_st2 = RF.RF_ST('./data/Normalized_Data/EVAL_DataSetST.csv', './data/Normalized_Data/sub_DataSetST.csv') gdbt_st1 = GDBT_ST.GDBT_ST('./data/Normalized_Data/testing_DataSetST.csv', './data/Normalized_Data/sub_DataSetST.csv') gdbt_st2 = GDBT_ST.GDBT_ST('./data/Normalized_Data/EVAL_DataSetST.csv', './data/Normalized_Data/sub_DataSetST.csv') gdbt_st3 = GDBT_ST.GDBT_ST('./data/Normalized_Data/EVAL_DataSetST.csv', './data/Normalized_Data/sub_DataSetST.csv') svr_st = s.SVR_ST('./data/Normalized_Data/EVAL_DataSetST.csv', './data/Normalized_Data/sub_DataSetST.csv') fw = open(tmpFilename, 'a') #追加写入文件 for i in range(len(svr_st)): fw.write('%s,%s,%.4f,%.4f,%.4f,%.4f,%.4f,%.4f,%.4f,%.4f,%.4f,%.4f,%.4f\n' % (gdbt_st1[i][0],gdbt_st1[i][1], float(RF_st1[i][2]), float(RF_st2[i][2]), float(gdbt_st1[i][2]), float(gdbt_st2[i][2]), float(gdbt_st3[i][2]), float(svr_st[i][2]), float(svr_st[i][3]), float(svr_st[i][4]), (float(RF_st2[i][2])+float(gdbt_st1[i][2]) + float(gdbt_st2[i][2]) + \ float(svr_st[i][2])+float(svr_st[i][3])+float(svr_st[i][4])) / 6, cost_dict[gdbt_st1[i][1]][gdbt_st1[i][0]][0], cost_dict[gdbt_st1[i][1]][gdbt_st1[i][0]][1], )) fw.close()