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

最新推荐文章于 2024-09-02 14:39:58 发布

「已注销」

最新推荐文章于 2024-09-02 14:39:58 发布

阅读量6.6k

点赞数 2

分类专栏：机器学习

本文链接：https://blog.csdn.net/weixin_40937909/article/details/80154879

版权

在使用LSTM模型时，训练样本通常需要统一时间步长，但测试阶段可能遇到不同长度的序列。已训练好的LSTM模型可以通过调整模型的`batch_input_shape`来适应变长输入，例如将[null, 100, 5]中的100改为，允许不同长度的序列。这种方法可能也适用于模型的直接修改。" 109745277,10282934,使用Jsoup解析和操作HTML,"['java', 'html5', 'css', '数据提取', '网页解析']

摘要由CSDN通过智能技术生成

最近遇到个问题，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":