Commit 97e2c76d authored by Stefano Franchellucci's avatar Stefano Franchellucci Committed by Alexander Froch
Browse files

Fixing bug, missing tracks labels in preprocessed sample

parent 25d35d5f
...@@ -4,6 +4,10 @@ tracks_labels: ...@@ -4,6 +4,10 @@ tracks_labels:
- truthOriginLabel - truthOriginLabel
- truthVertexIndex - truthVertexIndex
tracks_loose_labels:
- truthOriginLabel
- truthVertexIndex
train_variables: train_variables:
JetKinematics: JetKinematics:
- eta_btagJes - eta_btagJes
......
...@@ -91,8 +91,8 @@ def apply_scaling_trks( ...@@ -91,8 +91,8 @@ def apply_scaling_trks(
var_arr_list.append(np.nan_to_num(x)) var_arr_list.append(np.nan_to_num(x))
# track vertex and origin labels # track vertex and origin labels
if "track_labels" in variable_config: if f"{tracks_name}_labels" in variable_config:
trkLabels = variable_config["track_labels"] trkLabels = variable_config[f"{tracks_name}_labels"]
trk_labels = np.stack( trk_labels = np.stack(
[np.nan_to_num(trks[v]) for v in trkLabels], [np.nan_to_num(trks[v]) for v in trkLabels],
axis=-1, axis=-1,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment