[WIP]

Author: abelsiqueira

src/convenience.jl

@@ -28,6 +28,9 @@ function cluster!(
   distance::SemiMetric = SqEuclidean(),
   weight_type::Symbol = :convex,
   tol::Float64 = 1e-2,
+  niters::Int = 100,
+  learning_rate::Float64 = 0.001,
+  adaptive_grad::Bool = false,
 )
   df = DuckDB.query(
     connection,
@@ -37,7 +40,7 @@ function cluster!(
   split_into_periods!(df; period_duration)
   clusters =
     find_representative_periods(df, num_rps; drop_incomplete_last_period, method, distance)
-  fit_rep_period_weights!(clusters; weight_type, tol)
+  fit_rep_period_weights!(clusters; weight_type, tol, niters, learning_rate, adaptive_grad)
 
   write_clustering_result_to_tables(connection, clusters)