From cdd693c397656787889b90d3b85c5111789ffb7a Mon Sep 17 00:00:00 2001 From: Yossi Synett Date: Wed, 24 Oct 2018 15:05:16 +0300 Subject: [PATCH] chapter 5 - fix bug in main score for multiple halos --- Chapter5_LossFunctions/DarkWorldsMetric.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Chapter5_LossFunctions/DarkWorldsMetric.py b/Chapter5_LossFunctions/DarkWorldsMetric.py index ce652277..dec8fa5f 100644 --- a/Chapter5_LossFunctions/DarkWorldsMetric.py +++ b/Chapter5_LossFunctions/DarkWorldsMetric.py @@ -205,8 +205,8 @@ def main_score( nhalo_all, x_true_all, y_true_all, x_ref_all, y_ref_all, sky_pre x_predicted=np.array([],dtype=float) y_predicted=np.array([],dtype=float) for i in range(nhalo): - x_predicted=np.append(x_predicted,float(sky[0])) #get the predicted values - y_predicted=np.append(y_predicted,float(sky[1])) + x_predicted=np.append(x_predicted,float(sky[2*i])) #get the predicted values + y_predicted=np.append(y_predicted,float(sky[2*i+1])) #The solution file for the test data provides masses #to calculate the centre of mass where as the Training_halo.csv #direct provides x_ref y_ref. So in the case of test data