Skip to content

Commit 53f01dc

Browse files
committed
Add optimal_active_set
1 parent 9118ee3 commit 53f01dc

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
*.jl.mem
44
/Manifest.toml
55
/docs/build/
6+
/test/profiling.jl
67

78
*.tar.gz
89
*.zip

src/regularized/regularized_generic.jl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,18 @@ function (regularized::RegularizedGeneric)(
5252
return dfw(θ; fw_kwargs=fw_kwargs)
5353
end
5454

55+
function optimal_active_set(
56+
regularized::RegularizedGeneric,
57+
θ::AbstractArray{<:Real};
58+
maximizer_kwargs=(;),
59+
fw_kwargs=(;),
60+
)
61+
(; f, ∇ₓf, maximizer, linear_solver) = regularized
62+
lmo = LMOWrapper(maximizer, maximizer_kwargs)
63+
dfw = DifferentiableFrankWolfe(f, ∇ₓf, lmo, linear_solver)
64+
return optimal_active_set(dfw, θ; fw_kwargs=fw_kwargs)
65+
end
66+
5567
## Backward pass, only works with vectors
5668

5769
function ChainRulesCore.rrule(

0 commit comments

Comments
 (0)