draft of interface manual struct conversion

Author: PyDataBlog

src/mlj_interface.jl

@@ -1,30 +1,59 @@
 # TODO 1: a using MLJModelInterface or import MLJModelInterface statement
+# Expose all instances of user specified structs and package artifcats.
+const ParallelKMeans_Desc = "Parallel & lightning fast implementation of all variants of the KMeans clustering algorithm in native Julia."
+
+# availalbe variants for reference
+const MLJDICT = Dict(:Lloyd => Lloyd(),
+                     :Hamerly => Hamerly(),
+                     :LightElkan => LightElkan())
 
 ####
 #### MODEL DEFINITION
 ####
 # TODO 2: MLJ-compatible model types and constructors
-@mlj_model mutable struct KMeans <: MLJModelInterface.Unsupervised
-    # Hyperparameters of the model
-    algo::Symbol                            = :Lloyd::(_ in (:Lloyd, :Hamerly, :LightElkan))
-    k_init::String                          = "k-means++"::(_ in ("k-means++", String)) # allow user seeding?
-    k::Int                                  = 3::(_ > 0)
-    tol::Float64                            = 1e-6::(_ < 1)
-    max_iters::Int                          = 300::(_ > 0)
-    copy::Bool                              = true
-    threads::Int                            = Threads.nthreads()::(_ > 0)
-    verbosity::Int                          = 0::(_ in (0, 1))  # Temp fix. Do we need to follow mlj verbosity style?
-    init = nothing
+
+mutable struct KMeans <: MLJModelInterface.Unsupervised
+    algo::Symbol
+    k_init::String
+    k::Int
+    tol::Float64
+    max_iters::Int
+    copy::Bool
+    threads::Int
+    verbosity::Int
+    init
 end
 
 
-# Expose all instances of user specified structs and package artifcats.
-const ParallelKMeans_Desc = "Parallel & lightning fast implementation of all variants of the KMeans clustering algorithm in native Julia."
+function KMeans(; algo=:Lloyd, k_init="k-means++",
+                k=3, tol=1e-6, max_iters=300, copy=true,
+                threads=Threads.nthreads(), verbosity=0, init=nothing)
+
+    model   = KMeans(algo, k_init, k, tol, max_iters, copy, threads, verbosity, init)
+    message = MLJModelInterface.clean!(model)
+    isempty(message) || @warn message
+    return model
+end
+
+
+function MLJModelInterface.clean!(m::KMeans)
+    warning = ""
+    if !(m.algo ∈ keys(MLJDICT))
+        warning *= "Unsuppored algorithm supplied. Please check documentation for supported Kmeans algorithms."
+    elseif m.k < 1
+        warning *= "Number of clusters must be greater than 0."
+    elseif !(m.tol < 1.0)
+        warning *= "Tolerance level must be less than 1."
+    elseif !(m.max_iters > 0)
+        warning *= "Number of permitted iterations must be greater than 0."
+    elseif !(m.threads > 0)
+        warning *= "Number of threads must be at least 1."
+    elseif !(m.verbosity ∈ (0, 1))
+        warning *= "Verbosity must be either 0 (no info) or 1 (info requested)"
+    end
+    return warning
+end
 
-# availalbe variants for reference
-const MLJDICT = Dict(:Lloyd => Lloyd(),
-                     :Hamerly => Hamerly(),
-                     :LightElkan => LightElkan())
 
 # TODO 3: implementation of fit, predict, and fitted_params of the model
 ####

test/test07_mlj_interface.jl

@@ -1,6 +1,7 @@
 module TestMLJInterface
 
 using ParallelKMeans
+using ParallelKMeans: KMeans
 using Random
 using Test
 using Suppressor