当前位置: 首页>>编程示例 >>用法及示例精选 >>正文


Python tf.data.experimental.SqlDataset用法及代码示例

Dataset 由 SQL 查询的结果组成。

继承自:Dataset

用法

tf.data.experimental.SqlDataset(
    driver_name, data_source_name, query, output_types
)

参数

  • driver_name 包含数据库类型的 0-D tf.string 张量。目前,唯一支持的值是'sqlite'。
  • data_source_name 一个 0-D tf.string 张量,包含连接到数据库的连接字符串。
  • query 一个 0-D tf.string 张量,包含要执行的 SQL 查询。
  • output_types tf.DType 对象的元组,表示 query 返回的列的类型。

属性

  • element_spec 此数据集元素的类型规范。
    dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
    dataset.element_spec
    TensorSpec(shape=(), dtype=tf.int32, name=None)

    如需更多信息,请阅读本指南。

SqlDataset 允许用户从 SQL 查询的结果集中读取数据。例如:

dataset = tf.data.experimental.SqlDataset("sqlite", "/foo/bar.sqlite3",
                                          "SELECT name, age FROM people",
                                          (tf.string, tf.int32))
# Prints the rows of the result set of the above query.
for element in dataset:
  print(element)

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.experimental.SqlDataset。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。