我是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函数的情况下执行此操作,但我认为会
好好利用它。
答案 0 :(得分:0)
tflearn.datautils.to_categorical
的参数必须是整数字符串,也就是说y
必须像[1,2,3,....]!