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)
Esempio n. 2
0
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()