SpTn-Model-only-1P / load_model.py
KtiyaKK's picture
Update 1P model architecture and generation
a03c99b
Raw History Blame Contribute Delete
301 Bytes
import numpy as np
MODEL_FILE = "sp_tn_1p.npz"
model = np.load(MODEL_FILE, allow_pickle=True)
w = model["w"]
vocab = model["vocab"]
print("SpTn-Model-only-1P")
print("Trainable parameters: 1")
print("Parameter:", float(w))
print("Vocabulary size:", len(vocab))
print("Model loaded successfully.")