lua - 什么是 DataSource 与 torch/dp 库的正确用法

Question

我对编程语言 lua 和火炬库都是新手。我正在尝试让一些机器学习算法尽快工作。我尝试使用此处的示例使用 dp 库获取神经网络。但是我无法将我的数据集放入表格中以输入学习算法。我认为我最好的也是最初的猜测是这样做：

train_set = dp.DataSet(dataset[1], dataset[2]) 
test_set = dp.DataSet(test_dataset[1], test_dataset[2])

ds = dp.DataSource(train_set=train_set, test_set=test_set)

这给出了错误： filename.lua:56: ')' expected near '='

其中 dataset[1] 是一个 torch.Tensor，其中包含有关数据的信息，而 dataset[2] 是我想知道的有关数据的二进制信息的 torch.Tensor。

希望这不是一个愚蠢的语法错误。

Answer 1

是的，这是一个语法错误。Lua 没有命名参数。Lua 专家使用 table 来模拟这样的功能。

所以，试试这个： dp.DataSource({train_set=train_set, test_set=test_set}) 或者只是 dp.DataSource{train_set=train_set, test_set=test_set} （如果一个函数有一个参数，你可以删除括号）。