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)
