Attempt to save more LRP data, in-out-forget LRP?

ichalkiad · Jul 27, 2017 · ca253d0 · ca253d0
1 parent 4b6aecb
commit ca253d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lrp.py b/src/lrp.py
@@ -337,7 +337,7 @@ def lrp_full(model,input_filename,net_arch,net_arch_layers,test_data_json,fc_out
          dstMat = get_DstMatrix_singleReview(save_dir+review_filename,test_data_json,k)
          neuronWords_jsons.append(review_filename)
          neuronWords_data[review_filename] = dstMat
-         w = collections.OrderedDict(words=kkeys,scores=np.sum(lstm_lrp_x,axis=1))
+         w = collections.OrderedDict(words=kkeys,scores=[np.sum(lstm_lrp_x,axis=1),np.sum(lstm_lrp_h,axis=1),np.sum(lstm_lrp_g,axis=1),np.sum(lstm_lrp_c,axis=1)])
          LRP[k] = w
 
     posNeg_predictionLabel = get_PosNegPredictionLabel(reviewLRP_data)