Skip to content
Snippets Groups Projects

clean scripts

1 file
+ 5
4
Compare changes
  • Side-by-side
  • Inline
@@ -52,13 +52,14 @@ class OmpForestRegressor(BaseEstimator):
:param X:
:return:
"""
D = self._forest_prediction(X)
forest_predictions = self._forest_prediction(X)
if self._models_parameters.normalize_D:
D /= self._forest_norms
forest_predictions /= self._forest_norms
# TODO: use self._models_parameters.normalize_weights here?
predictions = self._omp.predict(D)
predictions = self._omp.predict(forest_predictions) * (1 / (np.sum(self._omp.coef_) / len(np.nonzero(self._omp.coef_)))) \
if self._models_parameters.normalize_weights \
else self._omp.predict(forest_predictions)
return predictions
Loading