TestSVD.java

package org.shanbo.feluca.cf.test;

import java.util.Properties;


import org.shanbo.feluca.cf.common.Evaluation;
import org.shanbo.feluca.cf.common.Recommender;
import org.shanbo.feluca.cf.stars.factorization.RSVDModel;
import org.shanbo.feluca.cf.stars.factorization.SVDModel;
import org.shanbo.feluca.data2.DataEntry;
import org.shanbo.feluca.data2.DataEntry.RAMDataEntry;
import org.shanbo.feluca.data2.RandomAccessData;

public class TestSVD {

	/**
	 * @param args
	 * @throws Exception 
	 */
	public static void main(String[] args) throws Exception {

		String train = "data/mltrain";
		String test = "data/mltest";
		
		RAMDataEntry traind = new RAMDataEntry(train);
		
//		Recommender model = new SVDModel();
		Recommender model = new RSVDModel();
		
		Properties p = new Properties();
		p.setProperty("alpha", "0.006");
		p.setProperty("lambda", "0.015");
		p.setProperty("loops", "10");
		p.setProperty("factor", "50");
		p.setProperty("convergence", "0.90");
//		p.setProperty("-i", "true");
		model.setProperties(p);
		model.loadData(traind);
		model.train();
		
		
		DataEntry testd = DataEntry.createDataEntry(test, false);
		
		
		
		double result = Evaluation.runRMSE( testd, model);
		
		System.out.println("\n----\nRMSE:\t" + result);
		
	}

}