ANL-CEEESA/MIPLearn.jl
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.jl.git synced 2025-12-07 16:58:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes

Browse Source
This commit is contained in:
Alinson S. Xavier
2024-02-01 13:13:10 -06:00
parent d69c4bbfa7
commit 4d5b7e971c
3 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/src/components/test_cuts.jl
View File

@@ -28,15 +28,13 @@ end
function test_cuts()
data_filenames = ["$BASEDIR/../fixtures/stab-n50-0000$i.pkl.gz" for i in 0:0]
clf = pyimport("sklearn.neighbors").KNeighborsClassifier(n_neighbors=1)
clf = pyimport("sklearn.dummy").DummyClassifier()
extractor = H5FieldsExtractor(
instance_fields=["static_var_obj_coeffs"],
)
comp = MemorizingCutsComponent(clf=clf, extractor=extractor)
solver = LearningSolver(components=[comp])
solver.fit(data_filenames)
@show comp.n_features_
@show comp.n_targets_
stats = solver.optimize(
data_filenames[1],
data -> build_stab_model_jump(data, optimizer=SCIP.Optimizer),