import numpy as np
from sklearn.externals import joblib
from sklearn.linear_model import LogisticRegression

#X=np.loadtxt("Density_Xray_exist.dat")
X=np.load("Density_Xray_exist.npy")
y=np.array(X[:,2],dtype='int')
X = X[:,0:2]
X[:,1] = np.log10(X[:,1])

lr = LogisticRegression()
lr.fit(X,y)

joblib.dump(lr, 'X_ray_sources.pkl') 

clf = joblib.load('X_ray_sources.pkl') 


