Add information about names/levels to MCA struct

JuliaStats · Sep 4, 2024 · b657e15 · b657e15
1 parent 87b0bff
commit b657e15
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/mca.jl b/src/mca.jl
@@ -340,6 +340,9 @@ mutable struct MCA{T<:Real} <: LinearDimensionalityReduction
     # Variable names
     vnames::Vector{String}
 
+    # Variable+level names
+    names::NamedTuple
+
     # Map values to integer positions
     rd::Vector{Dict}
 
@@ -473,7 +476,9 @@ function fit(
 
     una, ben, gra = get_eigs(C.I, J, K)
 
-    mca = MCA(C, XI, vnames, rd, dr, K, J, una, ben, gra)
+    na = expand_names(vnames, dr)
+
+    mca = MCA(C, XI, vnames, na, rd, dr, K, J, una, ben, gra)
 
     return mca
 end