Keras 在AWS sagemaker上以脚本模式训练自动编码器

我想用keras训练一个自动编码器，其中X_列是mxn矩阵，y_列也是mxn矩阵。 例如枫树

X_train = np.array(([1, 2], 
               [3, 4]))
y_train = np.array(([5, 6], 
               [7, 8]))

我将两个矩阵串联在

训练集中

并保存到一个文件

training.npy

train_set = np.concatenate([X_train, y_train], axis=1)
print(train_set)
array([[1, 2, 5, 6],
       [3, 4, 7, 8]])

稍后我将其保存到S3

training_path_input = sess.upload_data('/tmp/training.npy', key_prefix=prefix+'/training')

现在，当我适合这个模型

model.fit({'train': training_path_input })

我想知道估计器将如何找到

X\u列

和

y\u列

的索引，因为

y\u列

与其他情况不同，不是一个向量。有没有办法在

fit（）

方法中指定这一点

---

或者有其他方法吗？

fit方法做两件事：（1）将数据从

training\u path\u input

（在S3上）复制到SageMaker培训实例中的

/opt/ml/input/data/

（在您的例子中是

/opt/ml/input/data/train

）和（2）使用您指定的任何超参数启动代码。您需要确保培训代码知道如何读取要复制到机器上的文件类型。您的培训代码必须包括将在本地读取复制文件的代码