web-dev-qa-db-ja.com

Tensorflow 2.0データセットとデータローダ

私はPytorchのユーザーで、私はPytorchのData.DataSetとData.Dataloader APIに慣れています。私はTensorflow 2.0で同じモデルを構築しようとしています、そして、私はPytorchのこれらのAPIと同様に機能するAPIがあるか疑問に思います。

そのようなAPIがない場合は、Tensorflowでデータロード部品を実装するために通常どのようにするかを教えてください。私はTensorflow 1を使用しましたが、DataSet APIの経験をしなかった。私は前に難しいです。オーバーライドのようなものがあることを願っていますgetItem入力としてのインデックスだけでインデックスだけです。

あまり前もってありがとう。

10
piljae.chae

私はPytorchに慣れていませんが、Tensorflowは次のシーケンスクラスを持つKeras APIを実装しています。

データセットなどの一連のデータにフィットするための基本オブジェクト

https://www.tensorflow.org/api_docs/python/tf/keras/utils/Sequence

このクラスには、インデックスのgetItemが含まれています。

3
Eirik Moseng