Tensorflow.js tf.data.CSVDataset 类 .columnNames() 方法

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

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