如何使用tflearn.data_utils.to_categorical(y,nb_classes)

时间:2018-06-24 00:39:03

标签: tflearn

我是python和tflearn的新手。 我被TFlearn的to_categorical函数所困扰。我阅读了文档,以下是网站上的内容:

tflearn.data_utils.to_categorical(y,nb_classes)

将类向量(从0到nb_class的整数)转换为二进制类矩阵,以用于categorical_crossentropy。

参数 y:数组。要转换的类向量。 nb_classes:整数。课程总数。

我的问题是,它实际上是作为参数接受的? 我有3个标签,A,B,C(熊猫系列)。我可以使用.iloc()。values将它们放入numpy数组中,以提供[A,C,B,C,C,B,A .... A]的大型数组 我只希望它们成为([1,0,0],[0,0,1],...)。 那么我实际上必须以哪种格式将它们作为to_categorical函数的参数传递?

非常感谢您的建议。谢谢
nb:可以在没有to_categorical函数的情况下执行此操作,但我认为会 好好利用它。

1 个答案:

答案 0 :(得分:0)

tflearn.datautils.to_categorical的参数必须是整数字符串,也就是说y必须像[1,2,3,....]!