LSTM的训练和测试长度不一样

  • Post author:
  • Post category:其他


最近遇到个问题,LSTM的训练的样本的时间长度是一致的,但是测试的时候用到生成器则希望其长度可以不定(根据生成的Latent space的长度不同而不同),搜了下好像没有特别好的方法(有的是训练的时候就改成timestep是

None

)但是我现在已经训练好了。

后来试了下去改model就可以了,

{
    "class_name": "Model",
    "config": {
        "name": "model_1",
        "layers": [{
            "name": "input_1",
            "class_name": "InputLayer",
            "config": {
                "batch_input_shape": [null, 100, 5],
                "dtype": "float32",
                "sparse": false,
                "name": "input_1"
            },
            "inbound_nodes": []
        }, {
            "name": "generator",
            "class_name": "Sequential",
            "config": [{
                "class_name": "LSTM",
                "config": {
                    "name": 



版权声明:本文为weixin_40937909原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。