回答編集履歴
1
d
answer
CHANGED
|
@@ -20,9 +20,9 @@
|
|
|
20
20
|
# 意図的にNoneを挿入
|
|
21
21
|
iris.data[0][0] = None
|
|
22
22
|
|
|
23
|
-
|
|
23
|
+
mask = np.isnan(iris.data).any(axis=1) # nan が含まれる行
|
|
24
|
-
data_wo_nan = iris.data[~
|
|
24
|
+
data_wo_nan = iris.data[~mask] # nan が含まれていない行だけ残す
|
|
25
|
-
labels_wo_nan = iris.target[~
|
|
25
|
+
labels_wo_nan = iris.target[~mask] # nan が含まれていない行だけ残す
|
|
26
26
|
|
|
27
27
|
# 決定木のモデル構築
|
|
28
28
|
clf = tree.DecisionTreeClassifier(max_depth=3)
|