FactCascade / example.py
nhnk's picture
Release frozen FactCascade utility router checkpoint and loading code
08bd580 verified
Raw History Blame Contribute Delete
385 Bytes
"""Load the trained router; X is the canonical ordered 31-feature matrix."""
from pathlib import Path
import numpy as np
from factcascade.utility_model import UtilityModel
root=Path(__file__).parent
model=UtilityModel.load(root / "router.json")
# Replace this synthetic vector with the frozen feature implementation output.
X=np.zeros((1,31),dtype=np.float64)
print(model.predict(X))