https://en.wikipedia.org/wiki/Iris_flower_data_set
↑データ
https://www.youtube.com/watch?v=tNa99PG8hR8
↑参考サイト
test_idx = [0, 50, 100]
↑この配列の数字は何なのか、分かりません。分かる方教えてください
python
1import numpy as np 2from sklearn.datasets import load_iris 3from sklearn import tree 4iris = load_iris() 5test_idx = [0, 50, 100] 6 7#training data 8train_target = np.delete(iris.target, test_idx) 9train_data = np.delete(iris.data, test_idx, axis=0) 10 11test_target = iris.target[test_idx] 12test_data = iris.data[test_idx] 13 14clf = tree.DecisionTreeClassifier() 15clf.fit(train_data, train_target) 16 17print (test_target) 18print (clf.predict(test_data)) 19
質問者さん自身が「参考サイト」として挙げたビデオで、明確に説明しています。ビデオを見て確かめるのにも、時間がかかるんです。まったく、時間を無駄にしました。こういうことは二度としないでください。