def train(self, dataset):
model = ensemble.RandomForestRegressor(n_estimators=self.get_number_estimators())
independent_data = self.get_independent_variable_data(dataset)
dependent_data = self.get_dependent_variable_data(dataset)
trained_regression = model.fit(independent_data, dependent_data)
return TrainedSklearnModel(self, trained_regression)
def train(self, dataset):
model = linear_model.Ridge(alpha=self.get_alpha())
independent_data = self.get_independent_variable_data(dataset)
dependent_data = self.get_dependent_variable_data(dataset)
trained_regression = model.fit(independent_data, dependent_data)
return TrainedSklearnModel(self, trained_regression)
def train(self, dataset):
model = tree.DecisionTreeRegressor()
independent_data = self.get_independent_variable_data(dataset)
dependent_data = self.get_dependent_variable_data(dataset)
trained_regression = model.fit(independent_data, dependent_data)
return TrainedSklearnModel(self, trained_regression)
def train(self, dataset):
model = ensemble.GradientBoostingRegressor(learning_rate=self.get_learning_rate(), \
n_estimators=self.get_number_estimators(), \
max_depth=self.get_max_depth()
)
independent_data = self.get_independent_variable_data(dataset)
dependent_data = self.get_dependent_variable_data(dataset)
trained_regression = model.fit(independent_data, dependent_data)
return TrainedSklearnModel(self, trained_regression)
def train(self, dataset):
model = linear_model.BayesianRidge(n_iter=self.get_number_iterations(),
alpha_1=self.get_alpha_1(),
alpha_2=self.get_alpha_2(),
lambda_1=self.get_lambda_1(),
lambda_2=self.get_lambda_2())
independent_data = self.get_independent_variable_data(dataset)
dependent_data = self.get_dependent_variable_data(dataset)
trained_regression = model.fit(independent_data, dependent_data)
return TrainedSklearnModel(self, trained_regression)
def train(self, dataset):
model = svm.SVC(
C=self.get_penalty_parameter(),
kernel=self.get_kernel(),
shrinking=self.get_shrinking(),
)
independent_data = self.get_independent_variable_data(dataset)
dependent_data = self.get_dependent_variable_data(dataset)
trained_regression = model.fit(independent_data, dependent_data)
return TrainedSklearnModel(self, trained_regression)