Skip to content

Commit d7a201d

Browse files
committed
Remove nSD adjustment from unpack.
1 parent 16535f5 commit d7a201d

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

machine_learning_hep/processer.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -239,11 +239,6 @@ def unpack(self, file_index):
239239
print('I am sorry, I am dying ...\n \n \n')
240240
sys.exit()
241241
dfreco = selectdfquery(dfreco, self.s_reco_unp)
242-
listnsd_old = dfreco["nsd_jet"].values
243-
dfreco["nsd_jet_orig"] = listnsd_old
244-
c_new = [0 if nsd_old < 0 else nsd_old for nsd_old in listnsd_old]
245-
dfreco = dfreco.drop(["nsd_jet"], axis=1)
246-
dfreco["nsd_jet"] = c_new
247242
dfreco = pd.merge(dfreco, dfevt, on=self.v_evtmatch)
248243
isselacc = selectfidacc(dfreco.pt_cand.values, dfreco.y_cand.values)
249244
dfreco = dfreco[np.array(isselacc, dtype=bool)]
@@ -284,11 +279,6 @@ def unpack(self, file_index):
284279
if self.mcordata == "mc":
285280
treegen = uproot.open(self.l_root[file_index])[self.n_treegen]
286281
dfgen = treegen.pandas.df(branches=self.v_gen)
287-
listnsd_old_gen = dfgen["nsd_jet"].values
288-
dfgen["nsd_jet_orig"] = listnsd_old_gen
289-
c_new_gen = [0 if nsd_old < 0 else nsd_old for nsd_old in listnsd_old_gen]
290-
dfgen = dfgen.drop(["nsd_jet"], axis=1)
291-
dfgen["nsd_jet"] = c_new_gen
292282
dfgen = pd.merge(dfgen, dfevtorig, on=self.v_evtmatch)
293283
dfgen = selectdfquery(dfgen, self.s_gen_unp)
294284
dfgen[self.v_isstd] = np.array(tag_bit_df(dfgen, self.v_bitvar,

0 commit comments

Comments
 (0)