Tensorflow.js tf.data.CSVDataset 类

Tensorflow.js 是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

Tensorflow.js 拥有丰富的库集，可用于创建、开发和部署模型，但训练或创建模型所需的主要元素是数据。用于开发和训练模型的数据通常很大，并且通常被视为 CSV 文件。因此，TensorFlow Library 提供了一个名为CSVDataset类的库来处理 CSV 文件。这个tf.data.CSVDataset类扩展了tf.data.Dataset类。

句法：

tf.data.csv( source )

方法： tf.data.CSVDataset类有一个名为columnNames()函数的预定义方法，该函数有助于获取 CSV 文件的列名。只要数据中有列，就使用columnNames()函数检索它，否则当文件没有列名时会引发错误。可以解析为数字的值作为类型 number 发出，其他值被解析为字符串。

句法：

tf.data.csv( source ).columnNames()

参数：此方法具有单个参数，如上所述和如下所述。

• source：源是 CSV 文件所在的文件。它可以是文件的链接，也可以是系统中的文件位置。

返回值：它以数组的形式返回 CSV 文件的列名。

示例 1：考虑以下 CSV 文件

S No., UserName, Address, Mobile number, Item Ordered, Payment Type
1, Bertram, 59 Oak Valley St., [phone], hair clip, COD
2, Cecil, Toledo, OH 43612, [phone], toy soldier, PAYPAL
3, Clarence, Port Saint Lucie, [phone], book of jokes, CREDIT CARD
4, Clive, Warwick, RI 02886, [phone], sharpie, CREDIT CARD
5, Maureen Massillon, OH 44646, [phone], shoes, DEBIT CARD

可以使用以下代码检索列名。

Javascript

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// This is the source of the csv file
// It can be a link or the location of the File
const source = 'SampleData.csv'
 
async function run() {
 
   // Creating Dataset from the source
   const csvDataset = tf.data.csv(Source);
 
   // Retrieving the column names from the
   // dataset using columnNames function
   const ColumnNames = (await csvDataset.columnNames());
    
   // Printing the ColumnNames
   console.log(ColumnNames)
}
 
await run();

输出：

S No., UserName, Address, Mobile number, Item Ordered, Payment Type

示例 2：考虑以下 CSV 文件

S No, Name, Marks(out of 100)
1, Geek 1, 31
2, Geek 2, 65
3, Geek 3, 97
4, Geek 4, 75
5, Geek 5, 58

可以使用以下代码检索列名。

Javascript

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// This is the source of the csv file
// It can be a link or the location of the File
const source = 'sample.csv'
 
async function run() {
 
   // Creating Dataset from the source
   const csvDataset = tf.data.csv(Source);
 
   // Retrieving the column names from
   // the dataset using columnNames function
   const ColumnNames = (await csvDataset.columnNames());
    
   // Printing the ColumnNames
   console.log(ColumnNames)
}
 
await run();

输出：

S No,Name,Marks(out of 100)

参考： https://js.tensorflow.org/api/latest/#class:data.CSVDataset