Merge pull request #6658 from RasaHQ/sklearn

deprecate sklearn policy
RasaHQ · Sep 14, 2020 · 46aec4c · 46aec4c
2 parents 8e898c2 + ab10406
commit 46aec4c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/changelog/6658.removal.md b/changelog/6658.removal.md
@@ -0,0 +1 @@
+`SklearnPolicy` was deprecated. `TEDPolicy` is the preferred machine-learning policy for dialogue models.
diff --git a/rasa/core/policies/sklearn_policy.py b/rasa/core/policies/sklearn_policy.py
@@ -118,6 +118,11 @@ def __init__(
         self._train_params = kwargs
         self.zero_state_features = zero_state_features or defaultdict(list)
 
+        rasa.shared.utils.io.raise_deprecation_warning(
+            f"'{SklearnPolicy.__name__}' is deprecated and will be removed in "
+            "the future. It is recommended to use the 'TEDPolicy' instead."
+        )
+
     @staticmethod
     def _default_model() -> Any:
         return LogisticRegression(solver="liblinear", multi_class="auto")
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		`SklearnPolicy` was deprecated. `TEDPolicy` is the preferred machine-learning policy for dialogue models.