forked from jwasham/practice-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathiris.py
More file actions
34 lines (22 loc) · 701 Bytes
/
iris.py
File metadata and controls
34 lines (22 loc) · 701 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
31
32
33
34
import numpy as np
from sklearn import tree
from sklearn.datasets import load_iris
def main():
iris = load_iris()
print(iris.feature_names)
print(iris.target_names)
# print(iris.data[0])
# print(iris.target[0])
test_idx = [0, 50, 100]
train_target = np.delete(iris.target, test_idx)
train_data = np.delete(iris.data, test_idx, axis=0)
test_target = iris.target[test_idx]
test_data = iris.data[test_idx]
# print(test_target)
# print(test_data)
clf = tree.DecisionTreeClassifier()
clf.fit(train_data, train_target)
print('actual:', test_target)
print('predicted:', clf.predict(test_data))
if __name__ == '__main__':
main()