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)
