回答編集履歴
1
コード追記
test
CHANGED
@@ -1,3 +1,41 @@
|
|
1
1
|
`Dataset(`の戻り値はオブジェクト1つなのにX,Y2つ返ることを前提としたコードになっているためです。
|
2
2
|
|
3
3
|
`X,Y = Dataset(~`を`X = Dataset(~`に修正することで提示エラーは発生しなくなります。
|
4
|
+
|
5
|
+
```Python
|
6
|
+
|
7
|
+
import pandas as pd
|
8
|
+
|
9
|
+
import numpy as np
|
10
|
+
|
11
|
+
class Dataset():
|
12
|
+
|
13
|
+
def __init__(self, file, root_dir, small=False):
|
14
|
+
|
15
|
+
self.df = pd.read_csv(root_dir + file)
|
16
|
+
|
17
|
+
|
18
|
+
|
19
|
+
def __len__(self):
|
20
|
+
|
21
|
+
return len(self.df)
|
22
|
+
|
23
|
+
|
24
|
+
|
25
|
+
def __getitem__(self, id):
|
26
|
+
|
27
|
+
a = self.df["A"][id]
|
28
|
+
|
29
|
+
b = np.array(self.df.values[id][1])
|
30
|
+
|
31
|
+
return a, b
|
32
|
+
|
33
|
+
|
34
|
+
|
35
|
+
ds = Dataset( 'hoge.csv', "./")
|
36
|
+
|
37
|
+
X,Y = ds[0]
|
38
|
+
|
39
|
+
print(X,Y) # 1 2 など
|
40
|
+
|
41
|
+
```
|