python-3.x - 如何在 python 中为 sklearn dump_svmlight_file 指定功能名称？

Question

文档： https ://scikit-learn.org/stable/modules/generated/sklearn.datasets.dump_svmlight_file.html

svmlight 遵循数据格式：

<target> <feature:value> <feature:value>

有了数据：

a = [[1,2,3],[4,5,6]]
b = [8,9]

运行命令：

dump_svmlight_file(a,b,'test.txt')

输出以下内容：

8 0:1 1:2 2:3
9 0:4 1:5 2:6

我想知道是否有一种方法可以指定功能名称而不是让它从 0 开始递增，我希望得到类似以下的结果：

1 10:5 50:15 100:50
0 10:15 25:5 75:15
1 20:5 40:5 60:5

dump_svmlight_file 命令有这样的能力吗？

Answer 1

没有。dump_svmlight_file没有内置该选项。源代码

您可以使用参数指定功能名称是从 0 开始还是从 1 开始zero_based。 文档

我建议您不要尝试使用实际功能名称转储文件，这会不必要地增加文件的大小。而是将您的功能名称单独腌制，然后加入它们。