Tensorflow.js tf.data.CSVDataset.columnNames()用法及代码示例

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

columnNames() 方法在 tf.data.CSVDatset 类下。它返回 CSV 文件的所有列名。

用法:

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

参数：此方法有一个参数，如下所述：

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

返回值：它返回字符串列表，即 CSV 文件的列名。

下面的例子将演示这种方法。

范例1：在此示例中，我们将查找以下值的所有列名称。

Serial Number,Name,Order Id,Amount,Payment Mode
12141661A321,Geek 1,YP12164AZEA,1200,DEBIT CARD
12141661A322,Geek 2,ZSER15563VS,23324,COD
12141661A323,Geek 3,DSR5442HG6,322,CREDIT CARD
12141661A324,Geek 4,GF3467FSGTW,1890,PAYPAL
12141661A325,Geek 5,RSTYCBBJSST,141,COD

我们现在将使用 columnNames() 方法检索列名。

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 the Dataset from the source
   const csvDataset = tf.data.csv(Source);
   // Retrieving the column names from the
   // dataset using columnNames() method
   const columnNames = await csvDataset.columnNames();
    
   // Printing the columnNames
   console.log(columnNames)
}
await run();

输出：

Serial Number, Name, Order Id, Amount, Payment Mode

范例2：在此示例中，我们将查找以下值的所有列名称。

S No,Name,Height(cm),Weight(Kgs)
2,Geek 2,167,58
3,Geek 3,179,46
1,Geek 1,164,51
4,Geek 4,166,53
5,Geek 5,138,63

我们现在将使用 columnNames() 方法检索列名。

Javascript


// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
const source2 = 'sampleData2.csv'
async function run() {
   // Create the Dataset from the source
   const csvDataset = tf.data.csv(source2);
   // Retrieve the column names from the
   // dataset using method
   const ColumnNames = (await csvDataset.columnNames());
    
   // Printing the column names
   console.log(ColumnNames)
}
await run();

输出：

S No, Name, Height(cm), Weight(Kgs)

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

相关用法

注：本文由纯净天空筛选整理自magichat大神的英文原创作品 Tensorflow.js tf.data.CSVDataset class .columnNames() Method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。