sfaira.estimators.EstimatorKerasEmbedding.get_one_time_tf_dataset

EstimatorKerasEmbedding.get_one_time_tf_dataset(idx, mode, batch_size=None, prefetch=None)