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