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

Remove MPS from HDF5 file

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-18 17:56:03 -05:00
parent 9c05e9d66a
commit 5418a40a7d
8 changed files with 83 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
test/fixtures/knapsack.jl vendored
View File

@@ -5,46 +5,46 @@
using JuMP
using MIPLearn
function build_knapsack_model()
# Create standard JuMP model
Base.@kwdef struct KnapsackData
weights = [1.0, 2.0, 3.0]
prices = [5.0, 6.0, 7.0]
capacity = 3.0
end
function build_knapsack_model(data = KnapsackData())
model = Model()
n = length(weights)
n = length(data.weights)
@variable(model, x[1:n], Bin)
@objective(model, Max, sum(x[i] * prices[i] for i = 1:n))
@constraint(model, c1, sum(x[i] * weights[i] for i = 1:n) <= capacity)
@objective(model, Max, sum(x[i] * data.prices[i] for i = 1:n))
@constraint(model, c1, sum(x[i] * data.weights[i] for i = 1:n) <= data.capacity)
# Add ML information to the model
@feature(model, [5.0])
@feature(c1, [1.0, 2.0, 3.0])
@category(c1, "c1")
for i = 1:n
@feature(x[i], [weights[i]; prices[i]])
@category(x[i], "type-$i")
end
# # Add ML information to the model
# @feature(model, [5.0])
# @feature(c1, [1.0, 2.0, 3.0])
# @category(c1, "c1")
# for i = 1:n
# @feature(x[i], [weights[i]; prices[i]])
# @category(x[i], "type-$i")
# end
# Should store ML information
@test model.ext[:miplearn]["variable_features"]["x[1]"] == [1.0, 5.0]
@test model.ext[:miplearn]["variable_features"]["x[2]"] == [2.0, 6.0]
@test model.ext[:miplearn]["variable_features"]["x[3]"] == [3.0, 7.0]
@test model.ext[:miplearn]["variable_categories"]["x[1]"] == "type-1"
@test model.ext[:miplearn]["variable_categories"]["x[2]"] == "type-2"
@test model.ext[:miplearn]["variable_categories"]["x[3]"] == "type-3"
@test model.ext[:miplearn]["constraint_features"]["c1"] == [1.0, 2.0, 3.0]
@test model.ext[:miplearn]["constraint_categories"]["c1"] == "c1"
@test model.ext[:miplearn]["instance_features"] == [5.0]
# # Should store ML information
# @test model.ext[:miplearn]["variable_features"]["x[1]"] == [1.0, 5.0]
# @test model.ext[:miplearn]["variable_features"]["x[2]"] == [2.0, 6.0]
# @test model.ext[:miplearn]["variable_features"]["x[3]"] == [3.0, 7.0]
# @test model.ext[:miplearn]["variable_categories"]["x[1]"] == "type-1"
# @test model.ext[:miplearn]["variable_categories"]["x[2]"] == "type-2"
# @test model.ext[:miplearn]["variable_categories"]["x[3]"] == "type-3"
# @test model.ext[:miplearn]["constraint_features"]["c1"] == [1.0, 2.0, 3.0]
# @test model.ext[:miplearn]["constraint_categories"]["c1"] == "c1"
# @test model.ext[:miplearn]["instance_features"] == [5.0]
return model
end
function build_knapsack_file_instance()
model = build_knapsack_model()
instance = JuMPInstance(model)
data = KnapsackData()
filename = tempname()
save(filename, instance)
return FileInstance(filename)
MIPLearn.save_data(filename, data)
return FileInstance(filename, build_knapsack_model)
end