add extra comments in UnivariateFinite type def

JuliaAI · May 2, 2020 · 354f115 · 354f115
1 parent e78c1d6
commit 354f115
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/interface/univariate_finite.jl b/src/interface/univariate_finite.jl
@@ -3,8 +3,13 @@ abstract type NonEuclidean <: Dist.ValueSupport end
 
 const UnivariateFiniteSuper = Dist.Distribution{Dist.Univariate,NonEuclidean}
 
-struct UnivariateFinite{L,U,T<:Real} <: UnivariateFiniteSuper
-    decoder::CategoricalDecoder{L,U}
+# C - original type (eg, Char in `categorical(['a', 'b'])`)
+# U - reference type <: Unsigned
+# T - raw probability type
+# L - subtype of CategoricalValue, eg CategoricalValue{Char,UInt32}
+
+struct UnivariateFinite{C,U,T<:Real} <: UnivariateFiniteSuper
+    decoder::CategoricalDecoder{C,U}
     prob_given_class::LittleDict{U,T}
 end