我正在尝试设计一个决策树的决策面,该决策树是根据 Google Colaboratory 数据集中的特征对进行训练的。
示例代码:
import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from sklearn.inspection import DecisionBoundaryDisplay
iris = load_iris()
但是 colab 会产生如下错误,
ImportError: cannot import name 'DecisionBoundaryDisplay' from
'sklearn.inspection' (/usr/local/lib/python3.7/dist-packages/sklearn/inspection/__init__.py).
如何在 Google Colab 中解决这个问题?
DecisionBoundaryDisplay
已添加到sklearn v1.1
中。 目前,Colab 默认有 sklearn v1.0.2
,您无法升级,因为 sklv1.1
需要 python>=v3.8
,但 Colab 有 python3.7.15.
这里已经有一些关于升级 Colab 的 python 版本的帖子;从在 Google Colaboratory 中安装 Python 3.8 内核开始,并根据需要查看链接/相关问题。