Added metrics to cross-validation script.

Tássia Camões Araújo
1 parent c4db70ca
Showing 1 changed file with 7 additions and 3 deletions Show diff stats
src/examples/cross_validation.py
@@ -40,16 +40,20 @@ if __name__ == &#39;__main__&#39;:
     try:
         cfg = Config()
         rec = Recommender(cfg)
+        print "\nRecommender strategy: ",rec.strategy.description
         user = LocalSystem()
-        user.maximal_pkg_profile()
-
+        #user.app_pkg_profile(rec.items_repository)
+        user.no_auto_pkg_profile()
         begin_time = datetime.datetime.now()
         logging.debug("Cross-validation started at %s" % begin_time)
  
         metrics = []
         metrics.append(Precision())
         metrics.append(Recall())
-        validation = CrossValidation(0.3,10,rec,metrics)
+        metrics.append(F1())
+        metrics.append(Accuracy())
+        metrics.append(SimpleAccuracy())
+        validation = CrossValidation(0.3,10,rec,metrics,0.005)
         validation.run(user)
         print validation
...	...	@@ -40,16 +40,20 @@ if __name__ == '__main__':
40	40	try:
41	41	cfg = Config()
42	42	rec = Recommender(cfg)
	43	+ print "\nRecommender strategy: ",rec.strategy.description
43	44	user = LocalSystem()
44		- user.maximal_pkg_profile()
45		-
	45	+ #user.app_pkg_profile(rec.items_repository)
	46	+ user.no_auto_pkg_profile()
46	47	begin_time = datetime.datetime.now()
47	48	logging.debug("Cross-validation started at %s" % begin_time)
48	49
49	50	metrics = []
50	51	metrics.append(Precision())
51	52	metrics.append(Recall())
52		- validation = CrossValidation(0.3,10,rec,metrics)
	53	+ metrics.append(F1())
	54	+ metrics.append(Accuracy())
	55	+ metrics.append(SimpleAccuracy())
	56	+ validation = CrossValidation(0.3,10,rec,metrics,0.005)
53	57	validation.run(user)
54	58	print validation
55	59
...	...