diff --git a/load_estimation_model.lua b/load_estimation_model.lua index 423e587..6b2d44d 100644 --- a/load_estimation_model.lua +++ b/load_estimation_model.lua @@ -30,5 +30,5 @@ local X = data[{{},{2,-1}}] model = torch.load('estimation_model.dat') local myPrediction = model:forward(X) p_file:write('NAME,F1,F2,F3,F4\n') -p_file:write(name..','..tostring(myPrediction[1][1])..','..tostring(myPrediction[1][2])..','..tostring(myPrediction[1][3])..','..tostring(myPrediction[1][4])..'\n') +p_file:write(name..','..tostring(1000*myPrediction[1][1])..','..tostring(1000*myPrediction[1][2])..','..tostring(1000*myPrediction[1][3])..','..tostring(1000*myPrediction[1][4])..'\n') diff --git a/load_tracking_model.lua b/load_tracking_model.lua index 8eca3b6..f84aa20 100644 --- a/load_tracking_model.lua +++ b/load_tracking_model.lua @@ -39,6 +39,6 @@ model = torch.load('tracking_model.dat') local myPrediction = model:forward(X) p_file:write('NAME,F1,F2,F3,F4\n') for p=1, (#myPrediction)[1] do - p_file:write(names[p]..','..tostring(myPrediction[p][1])..','..tostring(myPrediction[p][2])..','..tostring(myPrediction[p][3])..','..tostring(myPrediction[p][4])..'\n') + p_file:write(names[p]..','..tostring(1000*myPrediction[p][1])..','..tostring(10008myPrediction[p][2])..','..tostring(1000*myPrediction[p][3])..','..tostring(1000*myPrediction[p][4])..'\n') end