-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathiris.py
More file actions
29 lines (23 loc) · 841 Bytes
/
iris.py
File metadata and controls
29 lines (23 loc) · 841 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
import sklearn
from sklearn import datasets
from sklearn.datasets import load_iris
iris = load_iris()
#splitting train and test datasets
from sklearn.model_selection import train_test_split
x,y,z,a = train_test_split(iris.data, iris.target, test_size=0.1)
#x is train_iris {all features values counting 90%}
#y is remaining test_iris {10% of features}
#z is train_target {All tables containing 90% of iris target}
#a is test_target {remaning 10% of iris target}
#calling decision tree classifier
from sklearn import tree
dsclf = tree.DecisionTreeClassifier()
#now training data with decision
trained = dsclf.fit(x, z)
#now time for prediction
output = trained.predict(y)
print(output)
#checking accuracy %age
from sklearn.metrics import accuracy_score
check_pct = accuracy_score(a,output)
print(check_pct)