給LSTM網絡訓練的數據添加標籤(三維數組的一維增加數據)

最近遇到一個問題,自己訓練好神經網絡,輸出的是特徵數,但是結果需要送到分類器中,分類器計算準確率需要標籤, 所以需要解決的問題是: 在生成LSTM網絡的batch數時,需要加上標籤,但是在神經網絡進行訓練時,需要去掉標籤,預測完成後需要把標籤列添回去。 生成帶標籤的time_step數據,維度如上所示,但是這其中第二維中的38維裏有一維是標籤,進入網絡時需要去掉這一列。本來X_data是一個lis
相關文章
相關標籤/搜索