Update GenerateData.py

Author: oliverchampion

utilities/data_simulation/GenerateData.py

@@ -192,7 +192,7 @@ def simulate_training_data(self, bvalues, SNR = (5,100), n = 1000000, Drange = (
         #data_sim = np.zeros([len(D), len(bvalues)])
         bvalues = np.array(bvalues)
         if type(SNR) == tuple:
-            noise_std = 1/SNR[1] + test[:,4] * (1/SNR[0] - 1/SNR[1])
+            noise_std = 1/SNR[1] + test[:,3] * (1/SNR[0] - 1/SNR[1])
             addnoise = True
         elif SNR == 0:
             addnoise = False