tl; dr; 我可以限制tf.data.Dataset
中元素的数量吗?
A有一个训练和评估循环,可以处理整个给定的dataset
。对于测试而言,这是不理想的,因为它要花很长时间才能遍历整个数据集。我可以通过创建Mock数据集或限制dataset
的元素数量来测试此代码,因此该代码仅经过(例如)前10个数据点。我该怎么做第二个?
谢谢
答案 0 :(得分:2)
仅从>>> 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)