Skip to content

Commit

Permalink
Keep the multiRegressor serial, for now
Browse files Browse the repository at this point in the history
  • Loading branch information
azzeddinetiba committed Sep 27, 2023
1 parent 7623bbe commit 060bcbc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rom_am/regressors/polynomialLassoRegressor.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ def train(self, input_data, output_data):
super().train(input_data, output_data)

self.regr_model = make_pipeline(
PolynomialFeatures(self.poly_degree), MultiOutputRegressor(LassoLarsIC(criterion=self.criterion), n_jobs=-1))
PolynomialFeatures(self.poly_degree), MultiOutputRegressor(LassoLarsIC(criterion=self.criterion),))
self.regr_model.fit(input_data.T, output_data.T)

self.nonzeroIds = []
Expand Down

0 comments on commit 060bcbc

Please sign in to comment.