as categorical

sadit · sadit · commit 299c7e29e032 · 2025-04-07T14:06:48.000-06:00
diff --git a/src/emodel.jl b/src/emodel.jl
@@ -59,6 +59,7 @@ function VectorModel(ent::EntropyWeighting, lw::LocalWeighting, voc::Vocabulary,
             comb::CombineWeighting=NormalizedEntropy(),
             minbatch=0, verbose=true
         )
+    labels = categorical(labels)
     @assert length(labels) == length(corpus)
     labels = categorical_labels(labels)
     n = length(labels)

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,7 @@ function VectorModel(ent::EntropyWeighting, lw::LocalWeighting, voc::Vocabulary,`
`59`	`59`	`comb::CombineWeighting=NormalizedEntropy(),`
`60`	`60`	`minbatch=0, verbose=true`
`61`	`61`	`)`
	`62`	`+ labels = categorical(labels)`
`62`	`63`	`@assert length(labels) == length(corpus)`
`63`	`64`	`labels = categorical_labels(labels)`
`64`	`65`	`n = length(labels)`