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)