From 7ba29368bfe50c10d8a2111b2e30272c7842d971 Mon Sep 17 00:00:00 2001 From: Yuan-Ru Lin Date: Sun, 24 Nov 2024 19:09:38 -0800 Subject: [PATCH] Initial commit for adding affinity propagation --- docs/ModelDescriptors.toml | 1 + docs/src/list_of_supported_models.md | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/ModelDescriptors.toml b/docs/ModelDescriptors.toml index 4865da222..02bb1b946 100644 --- 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 index e0dde1ba7..35e72ac91 100644 --- 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