限制tf.data.Dataset中的项目数

Question

tl; dr; 我可以限制tf.data.Dataset中元素的数量吗？

A有一个训练和评估循环，可以处理整个给定的dataset。对于测试而言，这是不理想的，因为它要花很长时间才能遍历整个数据集。我可以通过创建Mock数据集或限制dataset的元素数量来测试此代码，因此该代码仅经过（例如）前10个数据点。我该怎么做第二个？

谢谢

Answer 1

仅从>>> import gspread >>> from oauth2client.service_account import ServiceAccountCredentials >>> scope = ['https://spreadsheets.google.com/feeds'] >>> creds = ServiceAccountCredentials.from_json_keyfile_name('client_secret.json', scope) >>> client = gspread.authorize(creds) >>> file_id = '1rYQ84SIKTFUOTl1xyFkHMNLt9CW3tOMFbOelOvVwx1k' >>> url = "https://docs.google.com/spreadsheets/d/{0}".format(file_id) >>> doc = client.open_by_url(url) >>> sheet = doc.get_worksheet(0) >>> sheet.updated >>> 中获取固定数量的元素n的最简单方法是使用Dataset。例如：

Dataset.take(n)