質問編集履歴
1
コードの追加
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -1,4 +1,38 @@
|
|
|
1
1
|
特定の座標(x,y)の点が赤(ラベル値0)、青(ラベル値1)のどちらに分類されるのかを求めたいです。
|
|
2
2
|
|
|
3
3
|
問題:(x,y)=(5.5,4.0)の点は赤、青のどちらに分類されるか？
|
|
4
|
+
すみません。初めて質問させていただきます。
|
|
5
|
+
```ここに言語を入力
|
|
6
|
+
#必要なライブラリの読み込み。
|
|
7
|
+
%matplotlib inline
|
|
8
|
+
import matplotlib.pyplot as plt
|
|
9
|
+
import numpy as np
|
|
10
|
+
|
|
11
|
+
#分類対象となる平面内に分布した点を生成。
|
|
12
|
+
L = 10 # 領域の一辺の長さ
|
|
13
|
+
N = 50 # 点の数
|
|
14
|
+
B = L/2 # 市松模様の境界値
|
|
15
|
+
|
|
16
|
+
np.random.seed(1234)
|
|
17
|
+
X = np.random.rand(N, 2) * L
|
|
18
|
+
y = []
|
|
19
|
+
for i in range(N):
|
|
20
|
+
x_, y_ = X[i, :]
|
|
21
|
+
if x_ < B and y_ < B or x_ > B and y_ > B:
|
|
22
|
+
y.append(1)
|
|
23
|
+
else:
|
|
24
|
+
y.append(0)
|
|
25
|
+
|
|
26
|
+
colors = ['blue' if yi == 1 else 'red' for yi in y]
|
|
27
|
+
plt.scatter(X[:, 0], X[:, 1], c=colors)
|
|
28
|
+
|
|
29
|
+
#決定木モデルにより上のデータを学習し、決定木を可視化。
|
|
30
|
+
from sklearn.tree import DecisionTreeClassifier
|
|
31
|
+
from sklearn.tree import plot_tree
|
|
32
|
+
|
|
33
|
+
model = DecisionTreeClassifier(random_state=20)
|
|
34
|
+
model.fit(X, y)
|
|
35
|
+
plt.figure(figsize=(12, 6))
|
|
4
|
-
|
|
36
|
+
plot_tree(model, class_names=["red", "blue"], feature_names=['x', 'y'],
|
|
37
|
+
impurity=False, filled=True, fontsize=16);
|
|
38
|
+
```
|