首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >为什么LSTM不能减少损失

为什么LSTM不能减少损失
EN

Stack Overflow用户
提问于 2018-10-08 17:47:58
回答 2查看 2.3K关注 0票数 1

我用lstm来训练一个模型来预测股票价格,我使用了这样的多到一个模式:

代码语言:javascript
复制
                                    O 
                                    | 
                              O O O O 
                              | | | | 
                              O O O O 

我使用了keras框架来构建网络,但网络的建立似乎并不容易。

下面是我的python的lstm源代码:

代码语言:javascript
复制
def lstm_rls(num_in,num_out=1, batch_size=128, step=1,dim=1):
    model = Sequential()
    
    model.add(LSTM(
        1024,
        input_shape=(step, num_in),
        return_sequences=True))
    model.add(Dropout(0.2))
    
    model.add(LSTM(
        512,
        return_sequences=False))
    model.add(Dropout(0.2))
    
    model.add(Dense(1))
    model.add(Activation('linear'))
    model.compile(loss='mse', optimizer='rmsprop')
    return model

我训练了大约512次,但损失总是在7周左右。

1/512时代 3968/3968 ============================== - 4s 978 4s/步进损失: 48.6274 2/512时代 3968/3968 ============================== - 1s 220 1s/步进损失: 11.1913 3/512时代 3968/3968 ============================== -1S221us/步进损失: 6.8059 4/512时代 3968/3968 ============================== - 1s 220 1s/步进损失: 6.7905 5/512时代 3968/3968 ============================== -1S221us/步进损失: 6.8151 6/512时代 3968/3968 ============================== -1S219us/步进损失: 6.7907 7/512时代 3968/3968 ============================== - 1s 220 1s/步进损失: 6.8060 8/512时代 3968/3968 ============================== -1S221us/步进损失: 6.7824

...

509/512时代 3968/3968 ============================== - 1s 222 1s/步进损失: 6.7807 510/512时代 3968/3968 ============================== -1s223 1s /步进损失: 6.8199 511/512时代 3968/3968 ============================== - 1s 222 1s/步进损失: 6.7726 512/512时代 3968/3968 ============================== - 1s 222 1s/步进损失: 6.7715

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2018-10-08 20:09:07

我有你的图像标题问题(我有四个Gpu特斯拉k20)。训练LSTM需要很长时间(大约几天)。我觉得你的问题很自然。我使用的一种方法是增加批次大小。另一方面,批处理大小的增加与性能和内存使用有关。

票数 1
EN

Stack Overflow用户

发布于 2018-10-10 18:55:01

今天我看了一个关于调试技巧的讲座。它说,您可以分析代码的一种方法是:验证对象函数在小训练集上减少,而不是第一次。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/52707600

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档