Skip to content

A simple utility to transform a CSV file into a Torch7 Tensor

License

Notifications You must be signed in to change notification settings

willkurt/csv2tensor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

csv2tensor

A simple utility to transform a CSV file into a Torch7 Tensor

csv2tensor assumes that you are working with CSV files containing only headers and and numeric data.

You can use luarocks to install csv2tensor

\>\>luarocks install csv2tensor

basic usage is to simply load the csv file into a torch.Tensor. A list a column names is returned as a second value. Columns are ordered alphabetically (even if this is not the case in the file) unless include parameter is passed. If the include option is passed then the columns will be in the order they appear in that argument.

csv2tensor = require 'csv2tensor'

training_tensor, column_names = csv2tensor.load("training.csv")

csv2tensor also allow you to exclude certain columns by an optional parameter.

no_label_tensor, column_names = csv2tensor.load("training.csv",{exclude={"label"})

In addition you can also include only specific columns (as shown in the example this is useful if you have a label in your csv)

labels_tensor = csv2tensor.load("training.csv",{include={"label"})

Note: in it's current state csv2tensor assumes you have done all the hard work and have a very straightforward CSV file, NA values, non number values, etc will cause errors. But for straightforward, numberic csv files this should make things a bit easier.

About

A simple utility to transform a CSV file into a Torch7 Tensor

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages