def build_regressor_SKLearn(self, trainDataFile):
		processor = ATTPlainHitProcessor()
		regressor = ATTSkLearnHitRegressor(processor)
		
		(hits_training_values, Y) = regressor.collect_train_hits_from_file(trainDataFile)
		#print "Train Values: ", np.shape(hits_training_values), np.shape(Y)
		
		regressor.train(hits_training_values, Y)
		
		return regressor
Ejemplo n.º 2
0
def build_regressor_SKLearn():
    processor = ATTPlainHitProcessor()
    regressor = ATTSkLearnHitRegressor(processor)

    (hits_training_values, Y) = regressor.collect_train_hits_from_file(TRAIN_DATA_FILE)
    print "Train Values: ", np.shape(hits_training_values), np.shape(Y)

    regressor.train(hits_training_values, Y)

    return regressor
def build_regressor_2():
	processor = ATTPlainHitProcessor()
	regressor = ATTSkLearnHitRegressor(processor)
	
	TRAIN_VALUES_FILE_LEFT = "../data/train_points_20160129_left.txt"	
	(hits_training_values, Y) = regressor.collect_train_hits_from_file(TRAIN_VALUES_FILE_LEFT)
	print "Train Values: ", np.shape(hits_training_values), np.shape(Y)
	
	regressor.train(hits_training_values, Y)
	
	return regressor