From 234ddd3915c4f38f54b1b30ba1cbfdf1b1b65237 Mon Sep 17 00:00:00 2001 From: Guillaume Lemaitre Date: Fri, 6 Aug 2021 11:59:19 +0200 Subject: [PATCH] FIX correct some typo from 19859 --- sklearn/ensemble/_forest.py | 3 +++ sklearn/neural_network/_multilayer_perceptron.py | 3 +-- 2 files changed, 4 insertions(+), 2 deletions(-) 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):