Skip to content

Commit e0fe22e

Browse files
committedDec 7, 2016
added voting classifier
added voting classifier
1 parent 7de9fd7 commit e0fe22e

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed
 

‎titanic/titanic_kernel.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,13 @@
44
import numpy as np
55
import matplotlib.pyplot as plt
66

7-
# machine learning
87
from sklearn.linear_model import LogisticRegression
98
from sklearn.svm import SVC
109
from sklearn.ensemble import RandomForestClassifier
1110
from sklearn.neighbors import KNeighborsClassifier
11+
from sklearn.ensemble import VotingClassifier
12+
13+
np.random.seed(0)
1214

1315
if __name__ == "__main__":
1416

@@ -88,12 +90,17 @@
8890
knn.fit(X_train, y_train)
8991
score_knn = knn.score(X_train, y_train)
9092
out_knn = knn.predict(X_test)
91-
print "knn score: %f" %score_knn
93+
print "knn score: %f" %score_knn
94+
95+
#voting classifier
96+
vclf = VotingClassifier(estimators=[('rf',rfc),('lr',logreg),('svm',svc),('knn',knn)], voting='hard', weights=[2,1,2,1])
97+
vclf.fit(X_train, y_train)
98+
out_vclf = vclf.predict(X_test)
9299

93100
#write out predictions
94101
predictions_file = open("titanic_pred.csv", "wb")
95102
open_file_object = csv.writer(predictions_file)
96103
open_file_object.writerow(["PassengerId","Survived"])
97-
open_file_object.writerows(zip(idx, out_rfc))
104+
open_file_object.writerows(zip(idx, out_vclf.astype(int)))
98105
predictions_file.close()
99106

0 commit comments

Comments
 (0)