def testRidgeRegression(self): cwd = os.path.dirname(os.path.abspath(sys.argv[0])) data = LibsvmFileImporter(os.path.join(cwd,'data/regression/lin_reg'), binary=False).get_dataSet() rr = RidgeRegression(5) rr.trainModel(data) #TODO: create test self.assertTrue(True)
def regressionDemo(): '''how to use this modules as library''' # get data sets from file trainingSet = LibsvmFileImporter('./data/regression/lin_reg').get_dataSet() #testSet = LibsvmFileImporter('./data/lin_reg.t').get_dataSet() # make new ridge regression rr = RidgeRegression() for i in range(5): # optional: set complexity parameter rr.set_lambda(i) # 0: linear regression w/o regularization # train the model rr.trainModel(trainingSet) # get model print 'model:\n%s' % rr.get_model()