如何在Python中设置**参数

Question

我是Python的新手。

我正在使用Python 3.7.7和Tensorflow 2.1.0。

这是我的代码：

import tensorflow as tf
import tensorflow_datasets as tfds

dict = {"name": "omniglot:3.0.0", "data_dir": "d:\\tmp"}
omniglot_builder = tfds.builder("omniglot:3.0.0", builder_init_kwargs=dict)
omniglot_builder.download_and_prepare(download_dir="d:\\tmp")

但我收到此错误：

获得了意外的关键字参数'builder_init_kwargs'

我想设置data_dir，但我不知道该怎么做。我试图在download_dir中设置omniglot_builder.download_and_prepare(download_dir="d:\\tmp")，但仍将其下载到~/tensorflow_datasets。

摘自tdfs.builder的Tensorflow文档：

** builder_init_kwargs：传递给DatasetBuilder的关键字参数的字典。这些将覆盖在名称中传递的关键字参数，如果有。

如何设置builder_init_kwargs参数值？

Answer 1

要在python中设置kwargs参数，您只需在参数本身之前添加**。

因此，这将是您的代码：

import tensorflow as tf
import tensorflow_datasets as tfds

dict = {"name": "omniglot:3.0.0", "data_dir": "d:\\tmp"}
omniglot_builder = tfds.builder("omniglot:3.0.0", builder_init_kwargs=dict)
omniglot_builder.download_and_prepare(**download_dir="d:\\tmp")

当然，我只是在猜测，因为我知道kwargs的论点是什么，但我对tensorflow不熟悉。

希望这会有所帮助！