run lib fix

modneflib/modnef/modnef_torch/model.py

@@ -181,6 +181,8 @@ class ModNEFModel(nn.Module):
       output_spike, None
     """
 
+    print("hi")
+
     def to_aer(input):
       input = input.reshape(-1).to(torch.int32)
 
@@ -197,7 +199,7 @@ class ModNEFModel(nn.Module):
     batch_result = []
 
     for sample in input_spikes:
-      sample_res = self.driver.run_sample(sample, to_aer, True, len(self.layers))
+      sample_res = self.driver.run_sample(sample, to_aer, True, 2) #TODO
       batch_result.append([sample_res])
 
     return torch.tensor(batch_result).permute(1, 0, 2), None

modneflib/modnef/templates/run_lib.py

@@ -202,7 +202,7 @@ def fpga_evaluation(model, testLoader, board_path, driver_config, name="FPGA Eva
 
   return accuracy, y_pred, y_true   
 
-def confusion_matrix(y_true, y_pred, file_name, classes):
+def conf_matrix(y_true, y_pred, file_name, classes):
   cm = confusion_matrix(y_true, y_pred)
   df_cm = pd.DataFrame(cm / np.sum(cm, axis=1)[:, None], index = [i for i in classes], columns = [i for i in classes])
   plt.figure()