or-simplex-solvers/examples/simplex_visualization.py at main · melih545/or-simplex-solvers · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
"""Simplex visualization example."""

import os
import tempfile

cache_dir = os.path.join(tempfile.gettempdir(), "simplex-codes-matplotlib")
os.makedirs(cache_dir, exist_ok=True)
os.environ.setdefault("MPLBACKEND", "Agg")
os.environ.setdefault("MPLCONFIGDIR", cache_dir)

from simplex_codes.visualization import solve_and_plot_simplex


def main() -> None:
    result, axis = solve_and_plot_simplex(
        c=[-8, -5, 0, 0],
        A=[
            [1, 1, 1, 0],
            [9, 5, 0, 1],
        ],
        b=[6, 45],
        sense="max",
    )
    axis.figure.tight_layout()
    axis.figure.savefig("simplex-path.png", dpi=150)
    print("solution:", result.solution.tolist())
    print("objective_value:", result.objective_value)
    print("iterations:", result.iterations)
    print("saved:", "simplex-path.png")


if __name__ == "__main__":
    main()