在Google Colab中导入自定义库时出现问题

时间:2019-02-26 08:35:23

标签: python machine-learning computer-vision google-colaboratory

我正在做一些机器学习项目,并且想在google colab上运行该项目,因为我自己的机器太弱了,并且在尝试在其上运行项目时挂起。我的项目的结构如图所示。 project structure。 我有多个.py文件,每个文件都互相导入模块。我在PC中将项目转换为.zip文件,然后使用google colab上的Upload选项卡上传了项目。我解压缩了文件,并尝试从“ examples”文件夹中运行一个代码,该文件夹正在从modAL函数中导入某些函数,例如

from modAL.models import ActiveLearner.

此导入在Google colab上失败,错误为“没有名为modAL的模块”。有人可以告诉我如何解决这个问题吗?该代码可以在我自己的笔记本电脑上正常工作。

2 个答案:

答案 0 :(得分:0)

我找到了以下解释:https://zerowithdot.com/colab-workspace/-非常有用。 在Google驱动器中创建空间后

from os.path import join
from google.colab import drive
ROOT = "/content/drive"
drive.mount(ROOT)

获取git存储库

GIT_USERNAME = "xxx"
GIT_TOKEN = "xxx"
GIT_REPOSITORY = "Repo"

!mkdir "{PROJECT_PATH}"
!git clone https://{GIT_TOKEN}@github.com/{GIT_USERNAME}/{GIT_REPOSITORY}.git " 
{PROJECT_PATH}"

最后使用importlib来访问定义

from importlib.machinery import SourceFileLoader
somemodule = SourceFileLoader('somelib', join(PROJECT_PATH, 
'utils/somelib.py')).load_module()

答案 1 :(得分:0)

如果该项目是公共项目(可能也可以使其正常工作),则可以创建一个软件包[1]并使用pip安装它:

!pip install git+https://github.com/myuser/myproject

[1] https://packaging.python.org/tutorials/packaging-projects/