TensorFlowでcsvファイルからデータを読み取る方法

Question

csvからデータを読み取るサンプルコードを入手できますか？ 私の要件は、TensorFlowでCSVから列車とテストデータを生成する必要があります。

トレインデータとテストデータの両方を含む1つのCSV。私は列車のために最初の10行を意味し、次は試験のために10を意味します。 ありがとうございました

Answer 1

皆さんはTensorFlowでexcellent tutorialを作成しました。 csvから国勢調査データを読み込み、それをテンソルに変換し、高度な見積もりAPIを使用して機械学習モデルに適合させ評価する方法について説明します。

しかし、urllib関数を試したときにエラーが発生しました。コードを少し変更して、pandasを使って直接データを読み込むようにしました。

オリジナルコード

import tempfile 
import urllib 
train_file = tempfile.NamedTemporaryFile() 
test_file = tempfile.NamedTemporaryFile() 
urllib.urlretrieve("https://archive.ics.uci.edu/ml/machine-learning-databases/adult/adult.data", train_file.name) 
urllib.urlretrieve("https://archive.ics.uci.edu/ml/machine-learning-databases/adult/adult.test", test_file.name) 

import pandas as pd 
CSV_COLUMNS = [ 
    "age", "workclass", "fnlwgt", "education", "education_num", 
    "marital_status", "occupation", "relationship", "race", "gender", 
    "capital_gain", "capital_loss", "hours_per_week", "native_country", 
    "income_bracket"] 
df_train = pd.read_csv(train_file.name, names=CSV_COLUMNS, skipinitialspace=True) 
df_test = pd.read_csv(test_file.name, names=CSV_COLUMNS, skipinitialspace=True, skiprows=1) 

修正コード

import pandas as pd 
COLUMNS = ["age", "workclass", "fnlwgt", "education", "education_num", 
      "marital_status", "occupation", "relationship", "race", "gender", 
      "capital_gain", "capital_loss", "hours_per_week", "native_country", 
      "income_bracket"] 

df_train = pd.read_csv('http://mlr.cs.umass.edu/ml/machine-learning-databases/adult/adult.data' 
         , names=COLUMNS 
         , skipinitialspace=True) 
df_test = pd.read_csv('http://mlr.cs.umass.edu/ml/machine-learning-databases/adult/adult.test' 
         , names=COLUMNS 
         , skipinitialspace=True 
         , skiprows=1)