This commit is contained in:
Joseph Keshet
2016-07-12 21:46:38 -04:00
parent b46b4ada64
commit d531a5902d
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -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')
+1 -1
View File
@@ -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