import pandas as pd
import numpy as np
import Clustering as trk
import Fitting as fit
N = 1000
if __name__ == '__main__':
print("Reading file ...")
fitter = fit.TrackFitter(B=1.)
data_track = pd.DataFrame({
'event': [0],
'track': [0],
'pt': [0.],
'phi': [0.],
'xVtx': [0.],
'yVtx': [0.]
})
data_track = data_track.drop(data_track.index[[0]])
df = pd.read_csv("hits_" + str(N) + ".csv")
y_df = df[['particle']]
# X_df = df.drop(['hit','layer','particle','event'], axis=1)
X_df = df.drop(['particle', 'layer', 'iphi'], axis=1)
#replace particle with 100000000*event+particle
y_train = df['particle'].values + df['event'].values * 1000
