def load_data(self, filepath):
self.DataSet = DataSet.fromTrainTest(filepath)
mult = self.DataSet.trackword_config["TargetPrecision"]["full"] - \
self.DataSet.trackword_config["TargetPrecision"]["int"]
self.DataSet.X_train["b_trk_inv2R"] = self.DataSet.X_train["b_trk_inv2R"] * \
(2**(mult-self.DataSet.trackword_config["InvR"]["nbits"]))
self.DataSet.X_train["b_trk_cot"] = self.DataSet.X_train["b_trk_cot"] * \
(2**(mult-self.DataSet.trackword_config["Cot"]["nbits"]))
self.DataSet.X_train["b_trk_zT"] = self.DataSet.X_train["b_trk_zT"] * \
(2**(mult-self.DataSet.trackword_config["ZT"]["nbits"]))
self.DataSet.X_train["b_trk_phiT"] = self.DataSet.X_train["b_trk_phiT"] * \
(2**(mult-self.DataSet.trackword_config["PhiT"]["nbits"]))
# ==============================================================================
self.DataSet.X_test["b_trk_inv2R"] = self.DataSet.X_test["b_trk_inv2R"] * \
(2**(mult-self.DataSet.trackword_config["InvR"]["nbits"]))
self.DataSet.X_test["b_trk_cot"] = self.DataSet.X_test["b_trk_cot"] * \
(2**(mult-self.DataSet.trackword_config["Cot"]["nbits"]))
self.DataSet.X_test["b_trk_zT"] = self.DataSet.X_test["b_trk_zT"] * \
(2**(mult-self.DataSet.trackword_config["ZT"]["nbits"]))
self.DataSet.X_test["b_trk_phiT"] = self.DataSet.X_test["b_trk_phiT"] * \
(2**(mult-self.DataSet.trackword_config["PhiT"]["nbits"]))
for k in range(4):
self.DataSet.X_test["b_stub_r_" + str(k)] = self.DataSet.X_test["b_stub_r_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["r"]["nbits"]))
self.DataSet.X_test["b_stub_phi_" + str(k)] = self.DataSet.X_test["b_stub_phi_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["phi"]["nbits"]))
self.DataSet.X_test["b_stub_z_" + str(k)] = self.DataSet.X_test["b_stub_z_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["z"]["nbits"]))
self.DataSet.X_test["b_stub_dPhi_" + str(k)] = self.DataSet.X_test["b_stub_dPhi_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["dPhi"]["nbits"]))
self.DataSet.X_test["b_stub_dZ_" + str(k)] = self.DataSet.X_test["b_stub_dZ_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["dZ"]["nbits"]))
# ==============================================================================
self.DataSet.X_train["b_stub_r_" + str(k)] = self.DataSet.X_train["b_stub_r_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["r"]["nbits"]))
self.DataSet.X_train["b_stub_phi_" + str(k)] = self.DataSet.X_train["b_stub_phi_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["phi"]["nbits"]))
self.DataSet.X_train["b_stub_z_" + str(k)] = self.DataSet.X_train["b_stub_z_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["z"]["nbits"]))
self.DataSet.X_train["b_stub_dPhi_" + str(k)] = self.DataSet.X_train["b_stub_dPhi_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["dPhi"]["nbits"]))
self.DataSet.X_train["b_stub_dZ_" + str(k)] = self.DataSet.X_train["b_stub_dZ_" + str(k)] * \
(2**(mult-self.DataSet.trackword_config["dZ"]["nbits"]))
def load_data(self, filepath):
self.DataSet = DataSet.fromTrainTest(filepath)
self.train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(
self.DataSet.X_train, label="trk_fake")
self.test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(
self.DataSet.X_test, label="trk_fake")
def load_data(self, filepath):
self.DataSet = DataSet.fromTrainTest(filepath)
self.dtrain = xgb.DMatrix(self.DataSet.X_train.to_numpy(),
label=np.ravel(self.DataSet.y_train))
self.dtest = xgb.DMatrix(self.DataSet.X_test.to_numpy(),
label=np.ravel(self.DataSet.y_test))
def load_data(self, filepath):
self.DataSet = DataSet.fromTrainTest(filepath)
print(self.DataSet)
