在 Google Colab 中导入 `pyearth`(`from pyearth import Earth` 错误)

Import `pyearth` in Google Colab (`from pyearth import Earth` error)

我需要从 Google Colab 上名为 pyearth 的库加载多元自适应回归样条 (MARS) 算法。这就是我想要做的:

# Import model from library
from pyearth import Earth

# Initialize model
reg = Earth()

但是,Google Colab 默认没有该库。当我尝试 import pyearth:

时出现以下错误提示

ModuleNotFoundError: No module named 'pyearth'

因此我尝试使用 !pip 安装它,但是,如下所示,它也不起作用。

# Instal `pyearth`
!pip install pyearth # Runs smoothly

# Import Earth
from pyearth import Earth

> ImportError: cannot import name 'Earth' from 'pyearth' (/usr/local/lib/python3.7/dist-packages/pyearth/__init__.py)

奇怪的是,import pyearth 确实有效。

This post 解决了一个非常相似的问题,但仍未解决。唯一可用的答案对我不起作用。

原来 pyearth 是一个地球科学图书馆。换句话说,pyearth 与多元自适应回归样条 (MARS) 无关。

有MARS算法的库是sklearn-contrib-py-earth。这是在 Google Colab 上导入它的方法:

# Clone repo
!pip install git+https://github.com/scikit-learn-contrib/py-earth@v0.2dev

# Import model
from pyearth import Earth