获取input_shape参数LSTM的形状

时间:2020-08-20 23:13:15

标签: python machine-learning keras deep-learning recurrent-neural-network

是否有一种方法可以自动获取LSTM中input_shape参数的形状,然后将该形状设置为input_shape参数。 我希望能够使递归神经网络根据数据的形状自动设置输入形状。 谢谢。

inline void *new (const void *_class, ...)
{
    const class_t *class = _class;
    void *p = calloc(1, class->size);

    assert(p);
    memcpy(p, &class, sizeof(class));

    ...
}

2 个答案:

答案 0 :(得分:0)

如果您知道要在其上训练/测试模型的数据,这将不是一个挑战。您只需从数据集中选择一个数据点。如果您的数据位于NumPy / Tensor / Pandas中,则可以使用x.shape()获得其形状。您不必担心批次大小,这是Keras会自动选择的。

答案 1 :(得分:0)

我通常使用input_shape=X_train.shape[1:]。那是假设您输入的形状正确,并且可以通过神经网络。