def testSetC(self): rankSVM = RankSVM() rankSVM.setC(100.0) rankSVM.learnModel(self.X, self.y) predY = rankSVM.predict(self.X) auc1 = Evaluator.auc(predY, self.y) rankSVM.setC(0.1) rankSVM.learnModel(self.X, self.y) predY = rankSVM.predict(self.X) auc2 = Evaluator.auc(predY, self.y) self.assertTrue(auc1 != auc2)
def testPredict(self): rankSVM = RankSVM() rankSVM.learnModel(self.X, self.y) predY = rankSVM.predict(self.X)
def testLearnModel(self): rankSVM = RankSVM() rankSVM.learnModel(self.X, self.y)