在 Amazon Sagemaker Jupyter notebook 中导入自定义模块

Question

我想在 Sagemaker 的 jupyter 笔记本中导入自定义模块。 尝试从 Untitled1.ipynb 导入我尝试了两种不同的结构。 第一个是：

在“包文件夹”中有文件“cross_validation.py”和“ init .py”。 已经尝试了以下命令：

from package import cross_validation
import package.cross_validation

第二个是

我已经编码了import cross_validation

在这两种情况下，导入时我都没有收到任何错误，但我无法在模块内使用该类，因为我收到错误名称Class_X is not defined

我也重新启动了笔记本，以防万一，它仍然无法正常工作。 我怎么能做到？

Answer 1

如果您还需要从父目录导入文件，您可以添加到路径中：

import os
import sys
module_path = "/home/ec2-user/SageMaker/{module_name}"
if module_path not in sys.path:
    sys.path.append(module_path)

然后你可以像在普通的 python 环境中一样从项目的根目录导入

Answer 2

您可以将__init__.py文件添加到您的package目录中以使其成为 Python 包。 然后，您将从 Jupyter 笔记本内的包中导入模块

/home/ec2-user/SageMaker
    -- Notebook.ipynb 
    -- mypackage
        -- __init__.py
        -- mymodule.py

Notebook.ipynb 的内容

from mypackage.mymodule import SomeClass, SomeOtherClass

感谢您使用 Amazon SageMaker！

Answer 3

如果您使用 SageMaker Studio，则需要注意路径。

在SageMaker Studio的笔记本中：

!df -h ，您将看到以下行：

127.0.0.1:/200005  8.0E  1.3G  8.0E   1% /root

!pwd将是：

/root

它与bash 终端不同：

在 bash 终端中：

df -h
127.0.0.1:/200005  8.0E  1.4G  8.0E   1% /home/sagemaker-user

pwd
/home/sagemaker-user

因此，外部文件路径“/home/sagemaker-user”将映射到 SageMaker Studio 下笔记本中的“/root”。

因此，模块路径将更改：

import sys
sys.path.append('/root/module_name')

详细信息可以参考这个链接