1 files changed, 13 insertions, 21 deletions
diff --git a/main.py b/main.py
index 6b04638..5fa9504 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,6 @@
 from misc import *
 
-clist=[0.001]
+clist=[0.0001, 0.005, 0.01, 0.05, 0.1, 0.5, 1, 10, 100, 1000]
 
 inr="a.rid"
 
@@ -40,37 +40,29 @@ for ite in range(10):
         put(rid,entries)
 
     optc=1
-    bcmc=-1
+    bcmc=folds
 
     for c in clist:
         print(c)
         oum="%d-%g.m" % (o,c)
         
-        acmc = -1;
+        acmc = 0;
         for a in range(folds):
             rid = "b%d.rid"%a
-            params = "-T -d -m %s -i %s -o %s -c %g --iter 1 > /dev/null" % (inm,rid,oum,c)
+            params = "-T -d -m %s -i %s -o %s -c %g --cg_iter 100" % (inm,rid,oum,c)
             train(params)
+            rid="a%d.rid"%a
             params = "-V -C -m %s -i %s -s" %(oum,rid)
-            ncmc=cmc(params)
-            if acmc==-1:
-                acmc=ncmc
-            else:
-                for b in range(100):
-                    acmc[b]+=ncmc[b]
-        
-        if bcmc==-1:
+            acmc+=cmc(params)
+
+        print(acmc/folds)
+
+        if bcmc>acmc:
             optc=c
-            bcmc=ncmc
-        else:
-            for b in range(100):
-                if bcmc[b]<ncmc[b]:
-                    optc=c
-                    bcmc=ncmc
-                    break
+            bcmc=acmc
 
-    
+    print("train with: %g" % optc)    
     oum="%d.m"% o
-    params = "-T -d -m %s -i %s -o %s -c %g --iter 1 > /dev/null" % (inm,inr,oum,optc)
+    params = "-T -d -m %s -i %s -o %s -c %g" % (inm,inr,oum,optc)
     train(params)