Fetch the number of training inputs from hyperparameter list

eic · Jan 12, 2024 · 6b7dae7 · 6b7dae7
1 parent c4310f3
commit 6b7dae7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/roman_pots/train_dense_neural_network.py b/benchmarks/roman_pots/train_dense_neural_network.py
@@ -98,7 +98,7 @@ def run_experiment(hyperparameters):
   training_RP_pos = pd.DataFrame()
   training_MC_mom = pd.DataFrame()
 
-  for i in range(1,num_training_inputs+1):
+  for i in range(1,int(hyperparameters.num_training_inputs)+1):
     temp_training_RP_pos = pd.read_csv(hyperparameters.input_files+str(i)+'.txt', delimiter='\t', header=None)
     training_RP_pos = pd.concat([training_RP_pos, temp_training_RP_pos], ignore_index=True)
     temp_training_MC_mom = pd.read_csv(hyperparameters.target_files+str(i)+'.txt', delimiter='\t', header=None)