update train

1 files changed, 6 insertions, 2 deletions

diff --git a/train.py b/train.py
index 607d4df..82480e6 100644
--- a/train.py
+++ b/train.py
@@ -1,11 +1,15 @@
 import ensemble
 import cross
+import time
 
 clist=[0.0001, 0.005, 0.01, 0.05, 0.1, 0.5, 1, 10, 100, 1000]
 inr="a.rid"
 tot=316
-folds=4
+folds=2
 
 for i in range(10):
     print("iter %d:" %i)
-    ensemble.onefold(inr,"%d.m"%i,316,clist,folds,"%d.m"%(i+1))
+    start = time.time();
+    cross.onefold(inr,"%d.m"%i,316,clist,folds,"%d.m"%(i+1))
+    elapsed = time.time()-start
+    print("elapsed: %g" % elapsed)