teratail header banner
teratail header banner
質問するログイン新規登録

回答編集履歴

1

補足追加

2020/03/30 09:58

投稿

magichan
magichan

スコア15898

answer CHANGED
@@ -24,4 +24,46 @@
24
24
  res = test_data['Survived'].reset_index().values
25
25
  print(res)
26
26
  ```
27
- となります
27
+ となります
28
+
29
+ ---
30
+ **補足**
31
+
32
+ ```Python
33
+ from sklearn import svm
34
+ from sklearn.metrics import accuracy_score
35
+ import pandas as pd
36
+ import numpy as np
37
+
38
+ #学習データとラベルを準備
39
+ train_data=pd.read_csv("train1.csv",index_col=0)
40
+ print(train_data)
41
+ train_label=pd.read_csv("train_label1.csv",index_col=0)
42
+ print(train_label)
43
+
44
+ #テストデータを準備
45
+ test_data = pd.read_csv("test1.csv",index_col=0)
46
+ print(test_data)
47
+
48
+ #アルゴリズムを指定
49
+ clf = svm.SVC(C=1, gamma=10)
50
+
51
+ #学習
52
+ clf.fit(train_data,train_label)
53
+
54
+ #テスト
55
+ test_label = clf.predict(test_data)
56
+
57
+ #テスト結果の表示
58
+ print("テストデータ:{0},予測ラベル:{1}".format(test_data,test_label))
59
+ print("正解率= {}".format(accuracy_score(train_label, test_label)))
60
+
61
+ #テストデータにテスト結果を結合
62
+ test_data['Survived'] = test_label
63
+
64
+ #提案1:単にCSVに吐き出したいならばこれで良い
65
+ test_data['Survived'].to_csv('out.csv')
66
+ #提案2:Indexと結果を結合した結果の配列を得たいのであればこうなる
67
+ data = test_data['Survived'].reset_index().values
68
+ print(data)
69
+ ```