allenai / s2_fos

Apache License 2.0
32 stars 2 forks source link

Model implementation #1

Closed cmwilhelm closed 3 years ago

cmwilhelm commented 3 years ago

Using the s2agemaker-template as a starting point, implements the train/eval/serve trifecta for a paper field of study classifier.

The initial model is a multi-label SVM operating over ngram TFIDF embedding of paper title + abstract text.

/cc @yoganandc