def setUp(self): """ Setup the test package """ self.df = pd.read_csv( resource_filename(testfm.__name__, 'data/movielenshead.dat'), sep="::", header=None, names=['user', 'item', 'rating', 'date', 'title']) self.random_okapi = RandomOkapi( host="igraph-01", user="******", okapi_jar_dir="okapi/jar/efe97a00d2a1b3f30dbbaddb3f3dd4c7/", okapi_jar_base_name= "okapi-0.3.2-SNAPSHOT-jar-with-dependencies.jar", hadoop_source="/data/b.ajf/hadoop1_env.sh") self.popularity = PopularityOkapi( host="igraph-01", user="******", okapi_jar_dir="okapi/jar/efe97a00d2a1b3f30dbbaddb3f3dd4c7/", okapi_jar_base_name= "okapi-0.3.2-SNAPSHOT-jar-with-dependencies.jar", hadoop_source="/data/b.ajf/hadoop1_env.sh") self.bpr = BPROkapi( host="igraph-01", user="******", okapi_jar_dir="okapi/jar/efe97a00d2a1b3f30dbbaddb3f3dd4c7/", okapi_jar_base_name= "okapi-0.3.2-SNAPSHOT-jar-with-dependencies.jar", hadoop_source="/data/b.ajf/hadoop1_env.sh")
# Tell me what models we want to evaluate models = [ RandomModel(), PopularityOkapi(hadoop_source="/data/b.ajf/hadoop1_env.sh", host="igraph-01", okapi_jar_dir="okapi/jar/", #host='54.72.18.118', user='******', #okapi_jar_dir='/Users/linas/devel/okapi/target/', #okapi_jar_base_name='okapi-0.3.2-SNAPSHOT-jar-with-dependencies.jar', #public_key_path='/Users/linas/.ssh/hack-okapi.pem' ), Popularity(normalize=False), BPROkapi(hadoop_source="/data/b.ajf/hadoop1_env.sh", host="igraph-01", okapi_jar_dir="okapi/jar/", #host='54.72.18.118', user='******', #okapi_jar_dir='/Users/linas/devel/okapi/target/', # #okapi_jar_base_name='okapi-0.3.2-SNAPSHOT-jar-with-dependencies.jar', #public_key_path='/Users/linas/.ssh/hack-okapi.pem' ) ] # Setup the environment evaluator = Evaluator() for m in models: m.fit(df) print m.get_name().ljust(50), print evaluator.evaluate_model(m, df)