Merge pull request #1147 from Yuan-Ru-Lin/add-affinity-propagation

Initial commit for adding affinity propagation
JuliaAI · Dec 17, 2024 · d24ed5c · d24ed5c
2 parents 2403de6 + 7ba2936
commit d24ed5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/ModelDescriptors.toml b/docs/ModelDescriptors.toml
@@ -5,6 +5,7 @@ ARDRegressor_MLJScikitLearnInterface = ["regression", "Bayesian_models"]
 AdaBoostClassifier_MLJScikitLearnInterface = ["classification"]
 AdaBoostRegressor_MLJScikitLearnInterface = ["regression"]
 AdaBoostStumpClassifier_DecisionTree = ["classification"]
+AffinityPropagation_MLJClusteringInterface = ["clustering"]
 AffinityPropagation_MLJScikitLearnInterface = ["clustering"]
 AgglomerativeClustering_MLJScikitLearnInterface = ["clustering", "static_models"]
 AutoEncoder_BetaML = ["dimension_reduction"]

diff --git a/docs/src/list_of_supported_models.md b/docs/src/list_of_supported_models.md
@@ -32,7 +32,7 @@ independent assessment.
 | ------- | ------------- | ------ | -------- | ----
 [BetaML.jl](https://github.com/sylvaticus/BetaML.jl) | - | DecisionTreeClassifier, RandomForestClassifier, NeuralNetworkClassifier, PerceptronClassifier, KernelPerceptronClassifier, PegasosClassifier, DecisionTreeRegressor, RandomForestRegressor, NeuralNetworkRegressor, MultitargetNeuralNetworkRegressor, GaussianMixtureRegressor, MultitargetGaussianMixtureRegressor, KMeansClusterer, KMedoidsClusterer, GaussianMixtureClusterer,  SimpleImputer,  GaussianMixtureImputer, RandomForestImputer, GeneralImputer, AutoEncoder | medium |
 [CatBoost.jl](https://github.com/JuliaAI/CatBoost.jl) | - | CatBoostRegressor, CatBoostClassifier | high |
-[Clustering.jl](https://github.com/JuliaStats/Clustering.jl) | [MLJClusteringInterface.jl](https://github.com/JuliaAI/MLJClusteringInterface.jl) | KMeans, KMedoids, DBSCAN, HierarchicalClustering | high² |
+[Clustering.jl](https://github.com/JuliaStats/Clustering.jl) | [MLJClusteringInterface.jl](https://github.com/JuliaAI/MLJClusteringInterface.jl) | KMeans, KMedoids, AffinityPropagation, DBSCAN, HierarchicalClustering | high² |
 [DecisionTree.jl](https://github.com/bensadeghi/DecisionTree.jl) | [MLJDecisionTreeInterface.jl](https://github.com/JuliaAI/MLJDecisionTreeInterface.jl) | DecisionTreeClassifier, DecisionTreeRegressor, AdaBoostStumpClassifier, RandomForestClassifier, RandomForestRegressor | high | 
 [EvoTrees.jl](https://github.com/Evovest/EvoTrees.jl) | - | EvoTreeRegressor, EvoTreeClassifier, EvoTreeCount, EvoTreeGaussian, EvoTreeMLE | medium | tree-based gradient boosting models
 [EvoLinear.jl](https://github.com/jeremiedb/EvoLinear.jl) | - | EvoLinearRegressor | medium | linear boosting models