Skip to content

Commit 8d8b3d4

Browse files
committed
Make scaling benchmark build feasible random models
1 parent bd2e28c commit 8d8b3d4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

benchmark/scripts/benchmark_scaling.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def build_model(n_vars: int, n_cons: int, density: float) -> Model:
2525
m = Model()
2626
x = m.add_variables(lower=0, name="x", coords=[range(n_vars)])
2727

28-
data = rng.normal(loc=0.0, scale=1.0, size=int(n_vars * n_cons * density))
28+
data = rng.lognormal(mean=0.0, sigma=2.0, size=int(n_vars * n_cons * density))
2929
rows = rng.integers(0, n_cons, size=data.size)
3030
cols = rng.integers(0, n_vars, size=data.size)
3131

@@ -38,10 +38,10 @@ def build_model(n_vars: int, n_cons: int, density: float) -> Model:
3838
vars_idx = cols[mask]
3939
lhs = sum(coeff * x.isel(dim_0=idx) for coeff, idx in zip(coeffs, vars_idx))
4040
rhs = abs(coeffs).sum() * 0.1
41-
m.add_constraints(lhs == rhs, name=f"c{i}")
41+
m.add_constraints(lhs >= rhs, name=f"c{i}")
4242

4343
obj_coeffs = rng.uniform(0.1, 1.0, size=n_vars)
44-
m.objective = np.dot(obj_coeffs, x)
44+
m.objective = (obj_coeffs * x).sum()
4545
return m
4646

4747

0 commit comments

Comments
 (0)