Esempio n. 1
0
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()