コンストラクタ init, len, getitemの流れについて

python
1import torch
2class MyDataset(torch.utils.data.Dataset):
3
4    def __init__(self, data_num, transform=None):
5        self.transform = transform
6        self.data_num = data_num
7        self.data = []
8        self.label = []
9        for x in range(self.data_num):
10            self.data.append(x) # 0 から (data_num-1) までのリスト
11            self.label.append(x%2 == 0) # 偶数ならTrue 奇数ならFalse
12
13    def __len__(self):
14        return self.data_num
15
16    def __getitem__(self, idx):
17        out_data = self.data[idx]
18        out_label =  self.label[idx]
19
20        if self.transform:
21            out_data = self.transform(out_data)
22
23        return out_data, out_label
24

のようなクラスにおいて、コンスタクタlenではdata_numを返していますが、この値は、どこにいくのですか？initのあとに定義されているのにinitが受け取っているのですか？getitemを含めて、どういう流れで処理されているのか教えていただけませんか？

quickquip

2019/08/29 04:35

質問の"コンストラクタ"は"メソッド"の間違いかと思われます。

行動規範の内容に同意します

回答1件

ベストアンサー

`名前()` という名前の関数について

__len__()、__getitem__() など __名前__() となっている関数は Python で規定されている特殊なメソッドです。

自作したクラスにこれらの関数を定義しておくことで、自作クラスでも len() で長さを取得したり、[] 演算子で値を取得できるようになります。
(C++ でいう演算子オーバロードに似ています。)

以下の例では、Vector という自作クラスに __len__() 及び __getitem__() を定義しています。

python
1class Vector:
2    def __init__(self, items):
3        self.items = items
4
5    def __len__(self):
6        return len(self.items)
7
8    def __getitem__(self, idx):
9        return self.items[idx]
10    
11    
12vec = Vector(["A", "B", "C"])
13
14print(len(vec))  # 3
15print(vec[1])  # B

プログラムの流れについて

どこにいくのですか？initのあとに定義されているのにinitが受け取っているのですか？getitemを含めて、どういう流れで処理されているのか教えていただけませんか？

プログラム中で使用される箇所がなければ、 __len__() や __getitem__() は呼ばれません。
コードの中で data = MyDataset() で MyDataset オブジェクトを作成し、len(data) や data[インデックス] を実行している箇所があれば、対応する関数が呼ばれます。

今回の場合、__len__() はデータセットのサンプル数、__getitem__() は指定したインデックスのデータ及びラベルを返すように実装されているようです。