當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。