-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathpipeline.py
More file actions
30 lines (22 loc) · 832 Bytes
/
pipeline.py
File metadata and controls
30 lines (22 loc) · 832 Bytes
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
from sklearn import datasets
iris = datasets.load_iris()
X = iris.data
y = iris.target
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.5)
# Decision tree
from sklearn import tree
my_classifier = tree.DecisionTreeClassifier()
my_classifier.fit(X_train, y_train)
predictions = my_classifier.predict(X_test)
print predictions
from sklearn.metrics import accuracy_score
print "Decision tree: ", accuracy_score(y_test, predictions)
# K Nearest Neighbor
from sklearn.neighbors import KNeighborsClassifier
my_classifier = KNeighborsClassifier()
my_classifier.fit(X_train, y_train)
predictions = my_classifier.predict(X_test)
print predictions
from sklearn.metrics import accuracy_score
print "K nearest neigghbor: ", accuracy_score(y_test, predictions)