diff --git a/sklearn/ensemble/_forest.py b/sklearn/ensemble/_forest.py index 82f5378d9d07b..d56a5e9856872 100644 --- a/sklearn/ensemble/_forest.py +++ b/sklearn/ensemble/_forest.py @@ -894,6 +894,9 @@ def predict_log_proba(self, X): return proba + def _more_tags(self): + return {"multilabel": True} + class ForestRegressor(RegressorMixin, BaseForest, metaclass=ABCMeta): """ diff --git a/sklearn/neural_network/_multilayer_perceptron.py b/sklearn/neural_network/_multilayer_perceptron.py index ce57d7376665a..35ccb83a658ba 100644 --- a/sklearn/neural_network/_multilayer_perceptron.py +++ b/sklearn/neural_network/_multilayer_perceptron.py @@ -6,7 +6,6 @@ # Jiyuan Qian # License: BSD 3 clause -from tkinter.tix import Tree import numpy as np from abc import ABCMeta, abstractmethod @@ -1252,7 +1251,7 @@ def predict_proba(self, X): return y_pred def _more_tags(self): - return {"multilabel": Tree} + return {"multilabel": True} class MLPRegressor(RegressorMixin, BaseMultilayerPerceptron):