diff --git a/toolkit.py b/toolkit.py index 51d7aaeb9bab726880cc89d85bb16581e4f2b42c..3abfb06186a8212b015e276a39671cc5c25a695a 100755 --- a/toolkit.py +++ b/toolkit.py @@ -808,7 +808,7 @@ class ClassificationProject(object): if "weights" in np_kwargs: errors = [] for left, right in zip(bins, bins[1:]): - indices = np.where((x >= left) & (x <= right))[0] + indices = np.where((x >= left) & (x < right))[0] sumw2 = np.sum(np_kwargs["weights"][indices]**2) content = np.sum(np_kwargs["weights"][indices]) errors.append(math.sqrt(sumw2)/content) @@ -1085,7 +1085,7 @@ if __name__ == "__main__": #optimizer_opts=dict(lr=100., decay=1e-6, momentum=0.9), earlystopping_opts=dict(monitor='val_loss', min_delta=0, patience=2, verbose=0, mode='auto'), - selection="lep1Pt<5000", # cut out a few very weird outliers + selection="1", branches = ["met", "mt"], weight_expr = "eventWeight*genWeight", identifiers = ["DatasetNumber", "EventNumber"],