| """ |
| Basic Decision Example with InvariantOne v1. |
| |
| Demonstrates loading the frozen model and performing a 4-option decision. |
| """ |
|
|
| from invariantone import InvariantOne |
|
|
|
|
| def main() -> None: |
| print("Loading InvariantOne v1.0.0 (D5-L4 / Nop=16 / Seed 42)...") |
| model = InvariantOne.from_pretrained("invariantone-v1", device="cpu") |
|
|
| state = "Pressure is rising above the safe operating range." |
| question = "What action should the controller take?" |
| options = [ |
| "Close the inlet valve", |
| "Increase feed pressure", |
| "Maintain the current state", |
| "Disable monitoring", |
| ] |
|
|
| print(f"\nState:\n {state}") |
| print(f"Question:\n {question}") |
| print("Options:") |
| for i, opt in enumerate(options): |
| print(f" [{i}] {opt}") |
|
|
| result = model.decide( |
| state=state, |
| question=question, |
| options=options, |
| ) |
|
|
| print("\nDecision Result:") |
| print(f" Selected Choice: {result.choice} (index {result.choice_index})") |
| print(f" Probabilities: {[round(p, 4) for p in result.probabilities]}") |
| print(f" Raw Scores: {[round(s, 4) for s in result.scores]}") |
| print(f" Latency: {result.latency_ms:.1f} ms") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
