質問編集履歴

1

コードの加筆

2018/04/12 09:23

投稿

asddsa
asddsa

スコア20

test CHANGED
File without changes
test CHANGED
@@ -11,3 +11,57 @@
11
11
 
12
12
 
13
13
  なぜでしょうか。よろしくお願いします。
14
+
15
+
16
+
17
+ 以下、コードです。
18
+
19
+
20
+
21
+ import numpy as np
22
+
23
+ from sklearn import svm
24
+
25
+
26
+
27
+ for x in range(0, 10):
28
+
29
+ auc = []
30
+
31
+ X = np.array([[-3,-2],
32
+
33
+ [-1,0],
34
+
35
+ [-4,2],
36
+
37
+ [3,1],
38
+
39
+ [4,-1],
40
+
41
+ [-1,0],
42
+
43
+ [-2,-5],
44
+
45
+ [3,5],
46
+
47
+ [10,1],
48
+
49
+ [0,1]])
50
+
51
+
52
+
53
+ y = np.array([0,0,0,1,1,0,0,1,1,1])
54
+
55
+ skf = StratifiedKFold(n_splits=3, random_state = 42)
56
+
57
+ for train, test in skf.split(X, y):
58
+
59
+ clf = svm.SVC(probability=True)
60
+
61
+ clf.fit(X[train], y[train])
62
+
63
+ predict = clf.predict_proba(X[test])[:,1]
64
+
65
+ auc.append(roc_auc_score(y[test], predict))
66
+
67
+ print(np.array(auc).mean())