✅ Fix float type

FluxML · Sep 28, 2024 · b2bafd2 · b2bafd2
1 parent 3bf0902
commit b2bafd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/encoders.jl b/src/encoders.jl
@@ -22,7 +22,7 @@ function ordinal_encoder_fit(X; featinds)
         (Set([Float32(i) for i in 1:length(feat_levels)]) == Set(feat_levels)) && continue
         # Compute the dict using the given feature_mapper function
         mapping_matrix[i] =
-            Dict{Any, AbstractFloat}(
+            Dict{eltype(feat_levels), Float32}(
                 value => Float32(index) for (index, value) in enumerate(feat_levels)
             )
     end