Create a classifier class to wrap our model

paul-tqh-nguyen / reuters_topic_labelling

Deep learning to automatically label Reuter's articles with the relevant topics.

1 stars 0 forks source link

Create a classifier class to wrap our model #3

Closed paul-tqh-nguyen closed 4 years ago

paul-tqh-nguyen commented 4 years ago

This classifier class should have our model as an attribute, have a .train() method, have a method to load the data, have methods to classify new text, etc.

It'll make it easier for non-experts to be able to deal our model without having to learn torch.

paul-tqh-nguyen commented 4 years ago

As of https://github.com/paul-tqh-nguyen/reuters_topic_labelling/commit/f5c74e57e6f35aaaba403828cbb6748eef1aa05b, this task is complete.