关于make_dataset

首页课程实战体系课手记专栏慕课教程

关于make_dataset

来源：5-3 feature_column使用

manbaneverout

2020-05-04

老师您好，如下代码，运行后train_dataset中应该有几个样本？

def make_dataset(data_df, label_df, epochs=10, shuffle = True, batch_size=32):
    dataset = tf.data.Dataset.from_tensor_slices((dict(data_df), label_df))
    if shuffle:
        dataset = dataset.shuffle(10000)
    dataset = dataset.repeat(epochs).batch(batch_size)
    return dataset
train_dataset = make_dataset(train_df, y_train, batch_size=3)

写回答

1回答

正十七

正十七

2020-05-04

已采纳

会有len(data_df) * epochs个样本，也就是 len(data_df) * epochs / batch_size个batch。

0

1

manbaneverout

非常感谢！

2020-05-04

共1条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

estimator.train(input_fn = lambda : make_dataset(train_df, y_train, epochs=100))

回答 1

关于stateful = True的几个问题

回答 1

关于mask的疑问

回答 1

pass

回答 1

关于TensorFlow的LinearClassifier的预测

回答 1

打开慕课网App查看更多内容