willkurt / csv2tensor

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Returning single column

willkurt opened this issue · comments

Below makes this pretty obvious, when only one column is returned just return a vector not a Nx1 tensor

dataset_outputs = csv2tensor.load("example-logistic-regression.csv",
{include={"brand"}})

dataset_outputs = dataset_outputs[{{},1}]

fixed in v0.1-3