johannespischinger / senti_anal

MIT License
2 stars 0 forks source link

write models/predict_model.py and test_predict_model #65

Closed michaelfeil closed 2 years ago

michaelfeil commented 2 years ago

idea:

Class Predict:
       __init__(hydra_config, path_to_chph):
               self.tokenizer = ..
               self.model = .. load(ckph)
       predict(string):
               tokenized = self.tokenizer(string)
               preds = self.model(tokenized)

use this class then in cloud function local / anywhere else.

michaelfeil commented 2 years ago

@still we have no tests.

michaelfeil commented 2 years ago

works