LSTM(核心ML)的处理顺序

时间:2017-10-03 11:01:53

标签: python swift neural-network coreml coremltools

我已经通过coremltools将带有LSTM的Caffe模型转换为CoreML。现在我正在尝试执行它。但是,我找不到处理整个序列的方法

np.ndarray( (7, #sequence
             1, # batch
             120, 1, 1)) #items dims

因为我找不到设置隐藏状态( LSTM_1_c_in )和初始历史记录( LSTM_1_h_in )和自动的首字母的方法在处理下一个序列项时使用先前的状态/结果。

通过手动设置 LSTM_1_c_in LSTM_1_h_in 从以前的输出中手动重启方法'预测',模型重新启动 LSTM_1_h_out 分别为LSTM_1_c_out

是否可以通过1次运行处理整个序列?

P.S。使用Swift的方式也是可以接受的。

1 个答案:

答案 0 :(得分:0)

例如,LSTM的num_outputs等于3。

  • 如果lstm获得初始历史记录并且状态为np.ndarray((1,1,3)),它将只处理一个序列元素。 这是我的错误

  • 如果lstm获得初始历史记录并且状态为np.ndarray((3)),它将处理整个序列(但它只返回处理最后一个元素而没有完整历史记录的结果)。