diff --git a/src/Clustering/Clustering_MLJ.jl b/src/Clustering/Clustering_MLJ.jl index 33ca4ec9..b6389181 100644 --- 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 index 1ada7960..d8f29773 100644 --- 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