def testSetLearners(self): rankBoost = RankBoost() rankBoost.setIterations(10) rankBoost.setLearners(20) rankBoost.learnModel(self.X, self.y) predY = rankBoost.predict(self.X)
def testPredict(self): rankBoost = RankBoost() rankBoost.learnModel(self.X, self.y) predY = rankBoost.predict(self.X) self.assertTrue( Evaluator.auc(predY, self.y) <= 1.0 and Evaluator.auc(predY, self.y) >= 0.0)
def testLearnModel(self): rankBoost = RankBoost() rankBoost.learnModel(self.X, self.y)
def testPredict(self): rankBoost = RankBoost() rankBoost.learnModel(self.X, self.y) predY = rankBoost.predict(self.X) self.assertTrue(Evaluator.auc(predY, self.y) <= 1.0 and Evaluator.auc(predY, self.y) >= 0.0)