def setUp(self): # initialize query self.test_num_features = 6 test_query = """ 4 qid:1 1:2.6 2:1 3:2.1 4:0 5:2 6:1.4 # highly relevant 1 qid:1 1:1.2 2:1 3:2.9 4:0 5:2 6:1.9 # bad 0 qid:1 1:0.5 2:1 3:2.3 4:0 5:2 6:5.6 # not relevant 0 qid:1 1:0.5 2:1 3:2.3 4:0 5:2 6:5.6 # not relevant """ self.query_fh = cStringIO.StringIO(test_query) self.queries = query.Queries(self.query_fh, self.test_num_features) self.query = self.queries['1'] # initialize listwise learner self.learner = ListwiseLearningSystem(self.test_num_features, "--init_weights 0,0,1,0,0,0 --delta 1.0 --alpha 0.01 --ranker " "ranker.ProbabilisticRankingFunction --ranker_args 3 --ranker_tie " "first --comparison comparison.ProbabilisticInterleaveWithHistory" " --comparison_args \"--history_length 10 --biased true\"")