update

dmlc · Feb 27, 2016 · 2b1539c · 2b1539c
1 parent 4ca4198
commit 2b1539c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/run.py b/src/run.py
@@ -8,8 +8,8 @@
 #traindata='~/works/data/yahoo_protobuf_train_4by500'
 #testdata='~/works/data/yahoo_protobuf_valid'
 
-it=100
-fly=24
+it=10
+fly=4
 dim=2048
 
 #mf
@@ -23,8 +23,8 @@
 
 #admf
 eta_reg=2e-2
-#validdata='~/works/data/netflix_protobuf_valid'
-validdata='~/works/data/yahoo_protobuf_valid'
+validdata='~/works/data/netflix_protobuf_valid'
+#validdata='~/works/data/yahoo_protobuf_valid'
 
 import os
 import sys
@@ -33,7 +33,7 @@
   for eta_reg in [5e-1]:
     for temp in [1e-1]:
       for gam in [1.0]:
-        for dim in [128]:
+        for dim in [16]:
           print './mf --alg %s --train %s --test %s --valid %s --nu %d --nv %d --eta %e --lambda %e --gam %f --result %s --iter %d --dim %d --fly %d --epsilon %f --tau %d --temp %e --mineta %e --eta_reg %e'%(alg,traindata,testdata,validdata,nu,nv,eta,lam,gam,'%s_dim%d'%(alg,dim),it,dim,fly,eps,tau,temp,mineta,eta_reg)
           sys.stdout.flush()
           os.system('./mf --alg %s --train %s --test %s --valid %s --nu %d --nv %d --eta %e --lambda %e --gam %f --result %s --iter %d --dim %d --fly %d --epsilon %f --tau %d --temp %e --mineta %e --eta_reg %e'%(alg,traindata,testdata,validdata,nu,nv,eta,lam,gam,'%s_dim%d'%(alg,dim),it,dim,fly,eps,tau,temp,mineta,eta_reg))