Skip to content

Commit

Permalink
Merge branch 'main' into ak4-jets-near
Browse files Browse the repository at this point in the history
  • Loading branch information
cmantill authored May 21, 2024
2 parents ee9c97c + 080c0e9 commit e82db51
Show file tree
Hide file tree
Showing 37 changed files with 715 additions and 688 deletions.
6 changes: 6 additions & 0 deletions src/HH4b/boosted/TrainBDT.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,8 +150,11 @@ def preprocess_data(
training_keys = train_keys.copy()
print("Training keys ", training_keys)

print("events dict ", events_dict.keys())

for key in training_keys:
if key not in events_dict:
print(f"removing {key}")
training_keys.remove(key)

print("Train keys ", training_keys)
Expand Down Expand Up @@ -1090,6 +1093,8 @@ def main(args):
else:
years = args.year

print(years)

X_train = OrderedDict()
X_test = OrderedDict()
y_train = OrderedDict()
Expand All @@ -1108,6 +1113,7 @@ def main(args):
samples_run3[year].pop(key)

for year in years:
print("loading ", year)
events_dict_years[year] = load_run3_samples(
args.data_path,
year,
Expand Down
Binary file not shown.

This file was deleted.

Large diffs are not rendered by default.

Loading

0 comments on commit e82db51

Please sign in to comment.