5 lines
157 B
Julia
5 lines
157 B
Julia
export onehot, onecold
|
|
|
|
onehot(label, labels) = [i == label for i in labels]
|
|
onecold(pred, labels = 1:length(pred)) = labels[findfirst(pred, maximum(pred))]
|