Correct typo in MLJ fitted_params

Solves #64 and #65
sylvaticus · Jan 11, 2024 · f6f504e · f6f504e
1 parent fd8ec00
commit f6f504e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Clustering/Clustering_MLJ.jl b/src/Clustering/Clustering_MLJ.jl
@@ -175,7 +175,7 @@ function MMI.fit(m::Union{KMeans,KMedoids}, verbosity, X)
     report=nothing
     return ((classes=assignedClasses,centers=representatives,distanceFunction=m.dist), cache, report)
 end
-MMI.fitted_params(model::Union{KMeans,KMedoids}, fitresult) = (centers=fitesult[2], cluster_labels=CategoricalArrays.categorical(fitresults[1]))
+MMI.fitted_params(model::Union{KMeans,KMedoids}, fitresult) = (centers=fitresult[2], cluster_labels=CategoricalArrays.categorical(fitresults[1]))
 
 # ------------------------------------------------------------------------------
 # Transform functions...

diff --git a/src/GMM/GMM_MLJ.jl b/src/GMM/GMM_MLJ.jl
@@ -322,7 +322,7 @@ function MMI.fit(m::GaussianMixtureClusterer, verbosity, X)
     report     = (res.ϵ,res.lL,res.BIC,res.AIC)
     return (fitResults, cache, report)
 end
-MMI.fitted_params(model::GaussianMixtureClusterer, fitresult) = (weights=fitesult.pₖ, mixtures=fitresult.mixtures)
+MMI.fitted_params(model::GaussianMixtureClusterer, fitresult) = (weights=fitresult.pₖ, mixtures=fitresult.mixtures)
 
 function MMI.fit(m::GaussianMixtureRegressor, verbosity, X, y)
     x  = MMI.matrix(X) # convert table to matrix