32 lines
925 B
Julia
32 lines
925 B
Julia
|
struct OneHotVector <: AbstractVector{Bool}
|
||
|
|
ix::UInt32
|
||
|
|
of::UInt32
|
||
|
|
end
|
||
|
|
|
||
|
|
Base.size(xs::OneHotVector) = (Int64(xs.of),)
|
||
|
|
|
||
|
|
Base.getindex(xs::OneHotVector, i::Integer) = i == xs.ix
|
||
|
|
|
||
|
|
Base.:*(A::AbstractMatrix, b::OneHotVector) = A[:, b.ix]
|
||
|
|
|
||
|
|
struct OneHotMatrix <: AbstractMatrix{Bool}
|
||
|
|
data::Vector{OneHotVector}
|
||
|
|
end
|
||
|
|
|
||
|
|
Base.size(xs::OneHotMatrix) = (Int64(length(xs.data[1])),length(xs.data))
|
||
|
|
|
||
|
|
Base.getindex(xs::OneHotMatrix, i::Int, j::Int) = xs.data[j][i]
|
||
|
|
|
||
|
|
Base.:*(A::AbstractMatrix, B::OneHotMatrix) = A[:, map(x->x.ix, B.data)]
|
||
|
|
|
||
|
|
Base.hcat(x::OneHotVector, xs::OneHotVector...) = OneHotMatrix([x, xs...])
|
||
|
|
|
||
|
|
onehot(l, labels) = OneHotVector(findfirst(labels, l), length(labels))
|
||
|
|
onehotbatch(ls, labels) = OneHotMatrix([onehot(l, labels) for l in ls])
|
||
|
|
|
||
|
|
onecold(y::AbstractVector, labels = 1:length(y)) =
|
||
|
|
labels[findfirst(y, maximum(y))]
|
||
|
|
|
||
|
|
onecold(y::AbstractMatrix, l...) =
|
||
|
|
squeeze(mapslices(y -> onecold(y, l...), y, 1), 1)
|