亚马逊圣人。 SKlearn估算器与Tensorflow估算器-为什么其中之一不存在requirements_file?

时间:2019-05-06 18:09:44

标签: python requirements amazon-sagemaker

我正在查看Amazon Sagemaker中两个估算器SKLearn和Tensorflow的定义:

SKLearn

Tensorflow

class sagemaker.sklearn.estimator.SKLearn(entry_point, framework_version='0.20.0', source_dir=None, hyperparameters=None, py_version='py3', image_name=None, **kwargs)

class sagemaker.tensorflow.estimator.TensorFlow(training_steps=None, evaluation_steps=None, checkpoint_path=None, py_version='py2', framework_version=None, model_dir=None, requirements_file='', image_name=None, script_mode=False, distributions=None, **kwargs)

Tensorflow带有requirements_file参数,而SKLearn没有。有什么理由吗?如何将requirements.txt添加到SKLearn估算器?

1 个答案:

答案 0 :(得分:1)

我有类似的用例,尽管this issue表示他们将支持它,但我发现如果您将requirements.txt文件保留在入口点文件之外,则会下载所需的依赖项。