例如,我有以下内容:[1,2,3,5],我想要进行热编码。通常的结果如下:
[1,0,0,0,0]
[0,1,0,0,0]
[0,0,1,0,0]
[0,0,0,0,1]
但是相比于那种方法,我想要一种有条件的独热编码只有两个类别。所有小于3的值都被赋值为1,而大于等于3的值则被赋值为0,就像这样:
[1,0]
[1,0]
[0,1]
[0,1]
我知道如何做第一个,但是我在第二个问题上遇到了困难。请问有人可以帮我吗?