Skip to content

Commit b27588b

Browse files
committed
Add the (initial version of) spec of AffinityPropagation
1 parent 8f0e10a commit b27588b

File tree

2 files changed

+37
-1
lines changed

2 files changed

+37
-1
lines changed

src/registry/Metadata.toml

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5939,6 +5939,42 @@
59395939
":reporting_operations" = "`()`"
59405940
":constructor" = "`RecursiveFeatureElimination`"
59415941

5942+
[Clustering.AffinityPropagation]
5943+
":input_scitype" = "`ScientificTypesBase.Table{<:AbstractVector{<:ScientificTypesBase.Continuous}}`"
5944+
":output_scitype" = "`ScientificTypesBase.Table{<:AbstractVector{<:ScientificTypesBase.Continuous}}`"
5945+
":target_scitype" = "`ScientificTypesBase.Unknown`"
5946+
":fit_data_scitype" = "`Tuple{ScientificTypesBase.Table{<:AbstractVector{<:ScientificTypesBase.Continuous}}}`"
5947+
":predict_scitype" = "`ScientificTypesBase.Unknown`"
5948+
":transform_scitype" = "`ScientificTypesBase.Table{<:AbstractVector{<:ScientificTypesBase.Continuous}}`"
5949+
":inverse_transform_scitype" = "`ScientificTypesBase.Table{<:AbstractVector{<:ScientificTypesBase.Continuous}}`"
5950+
":target_in_fit" = "`false`"
5951+
":is_pure_julia" = "`true`"
5952+
":package_name" = "Clustering"
5953+
":package_license" = "MIT"
5954+
":load_path" = "MLJClusteringInterface.AffinityPropagation"
5955+
":package_uuid" = "aaaa29a8-35af-508c-8bc3-b662a17a0fe5"
5956+
":package_url" = "https://github.com/JuliaStats/Clustering.jl"
5957+
":is_wrapper" = "`false`"
5958+
":supports_weights" = "`false`"
5959+
":supports_class_weights" = "`false`"
5960+
":supports_online" = "`false`"
5961+
":docstring" = "To be added"
5962+
":name" = "AffinityPropagation"
5963+
":human_name" = "affinity propagation clusterer"
5964+
":is_supervised" = "`false`"
5965+
":prediction_type" = ":unknown"
5966+
":abstract_type" = "`MLJModelInterface.Unsupervised`"
5967+
":implemented_methods" = [":clean!", ":fit", ":fitted_params", ":predict", ":transform"]
5968+
":hyperparameters" = "`(:damp, :maxiter, :tol, :preference, :metric)`"
5969+
":hyperparameter_types" = "`(\"Float64\", \"Int64\", \"Float64\", \"Union{Nothing, Float64}\", \"Distances.SemiMetric\")`"
5970+
":hyperparameter_ranges" = "`(nothing, nothing, nothing, nothing, nothing)`"
5971+
":iteration_parameter" = "`nothing`"
5972+
":supports_training_losses" = "`false`"
5973+
":reports_feature_importances" = "`false`"
5974+
":deep_properties" = "`()`"
5975+
":reporting_operations" = "`(:predict,)`"
5976+
":constructor" = "`nothing`"
5977+
59425978
[Clustering.HierarchicalClustering]
59435979
":input_scitype" = "`Tuple{ScientificTypesBase.Table{<:AbstractVector{<:ScientificTypesBase.Continuous}}}`"
59445980
":output_scitype" = "`ScientificTypesBase.Unknown`"

src/registry/Models.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ MLJBalancing = ["BalancedBaggingClassifier", "BalancedModel"]
1717
Imbalance = ["RandomOversampler", "SMOTENC", "TomekUndersampler", "ClusterUndersampler", "SMOTE", "SMOTEN", "ROSE", "RandomUndersampler", "ENNUndersampler", "BorderlineSMOTE1", "RandomWalkOversampler"]
1818
MLJTuning = ["TunedModel"]
1919
FeatureSelection = ["FeatureSelector", "RecursiveFeatureElimination"]
20-
Clustering = ["HierarchicalClustering", "DBSCAN", "KMeans", "KMedoids"]
20+
Clustering = ["HierarchicalClustering", "DBSCAN", "KMeans", "KMedoids", "AffinityPropagation"]
2121
EvoLinear = ["EvoSplineRegressor", "EvoLinearRegressor"]
2222
MLJText = ["TfidfTransformer", "CountTransformer", "BM25Transformer"]
2323
LightGBM = ["LGBMClassifier", "LGBMRegressor"]

0 commit comments

Comments
 (0)