delete solvers_pycall.jl

ACEsuit · Jun 22, 2023 · 9f4e744 · 9f4e744
1 parent f3e6efe
commit 9f4e744
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 28 deletions.
diff --git a/src/ACEfit.jl b/src/ACEfit.jl
@@ -1,8 +1,5 @@
 module ACEfit
 
-
-include("solvers_pycall.jl")
-
 include("bayesianlinear.jl")
 include("data.jl")
 include("assemble.jl")

diff --git a/src/solvers.jl b/src/solvers.jl
@@ -117,3 +117,30 @@ end
 function solve(solver::BLR, A, y)
     return bayesian_linear_regression(A, y; solver.kwargs...)
 end
+
+@doc raw"""
+SKLEARN_BRR
+"""
+struct SKLEARN_BRR
+    tol::Number
+    n_iter::Integer
+end
+
+SKLEARN_BRR(; tol = 1e-3, n_iter = 300) = SKLEARN_BRR(tol, n_iter)
+
+# solve(solver::SKLEARN_BRR, ...) is implemented in ext/
+
+@doc raw"""
+SKLEARN_ARD
+"""
+struct SKLEARN_ARD
+    n_iter::Integer
+    tol::Number
+    threshold_lambda::Number
+end
+
+function SKLEARN_ARD(; n_iter = 300, tol = 1e-3, threshold_lambda = 10000)
+    SKLEARN_ARD(n_iter, tol, threshold_lambda)
+end
+
+# solve(solver::SKLEARN_ARD, ...) is implemented in ext/
diff --git a/src/solvers_pycall.jl b/src/solvers_pycall.jl