-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsol3.py
More file actions
60 lines (57 loc) · 2.35 KB
/
sol3.py
File metadata and controls
60 lines (57 loc) · 2.35 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import pandas as pd
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score,precision_score,recall_score,f1_score #model evaluation
df=pd.read_csv('https://sololearn.com/uploads/files/titanic.csv')
df['male']=df['Sex']=='male'
X=df[['Pclass','male','Age','Siblings/Spouses','Parents/Children','Fare']].values
y=df['Survived'].values
model= LogisticRegression()
model.fit(X,y)
y_pred=model.predict(X)
print("accuracy",accuracy_score(y,y_pred))
print("precision",precision_score(y,y_pred))
print("recall",recall_score(y,y_pred))
print("score",f1_score(y,y_pred))
from sklearn.metrics import confusion_matrix
print(confusion_matrix(y,y_pred))
from sklearn.model_selection import train_test_split
df=pd.read_csv('https://sololearn.com/uploads/files/titanic.csv')
df['male']=df['Sex']=='male'
X=df[['Pclass','male','Age','Siblings/Spouses','Parents/Children','Fare']].values
y=df['Survived'].values
model= LogisticRegression()
X_train,X_test,y_train,y_test= train_test_split(X,y)
model.fit(X_train,y_train)
print(model.score(X_test,y_test))
y_pred=model.predict(X_test)
print("accuracy",accuracy_score(y_test,y_pred))
print("precision",precision_score(y_test,y_pred))
print("recall",recall_score(y_test,y_pred))
print("score",f1_score(y_test,y_pred))
X=[1,1],[2,2],[3,3],[4,4] #random test
y=[0,0,1,1]
print("X_train",X_train)
print('y_train', y_train)
#logistic regression treshold, sensitivyn specificity, ROC curve
from sklearn.metrics import precision_recall_fscore_support
sensitivity_score=recall_score
def specificity_score(y_true,y_pred):
p,r,f,s= precision_recall_fscore_support(y_true,y_pred)
return r[0]
df=pd.read_csv('https://sololearn.com/uploads/files/titanic.csv')
df['male']=df['Sex']=='male'
X=df[['Pclass','male','Age','Siblings/Spouses','Parents/Children','Fare']].values
y=df['Survived'].values
model= LogisticRegression()
X_train,X_test,y_train,y_test= train_test_split(X,y)
model.fit(X_train,y_train)
print(model.score(X_test,y_test))
y_pred=model.predict(X_test)
print("sensitivity:",sensitivity_score(y_test,y_pred))
print("specificity:",specificity_score(y_test,y_pred))
#adjusting with proba
model.predict_proba(X_test)[:,1]>0.75
print("model proba:")
print(model.predict_proba(X_test))
print("precision",precision_score(y_test,y_pred))
print("recall",recall_score(y_test,y_pred))