aitech-eks-pub/wyk/pytorch_regression/linear1-infer.py
2021-05-19 13:21:13 +02:00

24 lines
387 B
Python
Executable File

#!/usr/bin/env python3
import sys
from analyzer import vectorizer
import torch
w = torch.load('model.bin')
def fun(w, x):
return x @ w
for line in sys.stdin:
line = line.strip('\n')
content = line
x = (torch.from_numpy(vectorizer.fit_transform([content]).toarray()))[0]
y_hat = fun(w, x)
print(max(1814.0, min(2013.999, y_hat.item() * 100.0 + 1913.0)))