MattStammers/Covid19_Text_Model
By MattStammers · ↓ 1 · ♥ 0 · Updated 2023-10-08
What's in this repo
This page lists the files in MattStammers/Covid19_Text_Model, including model.onnx.
- 1 ONNX models (.onnx files)
- 15 files, including 6 .json, 2 .pt, 2 .bin, 1 .md, 1 .onnx, 1 .pth
- Last updated 2023-10-08
ONNX files
| File | Size | Download |
|---|---|---|
| model.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
This is a basic inference BERT model which has been fine-tuned to discriminate between covid19 and non-covid-19 relevant texts.
Unlike past models I have created this one raw and uploaded it as a standalone git repo to experiment with upload options. Not as streamlined as using the Huggingface card generation system but definitely simpler to do.
This is also my first experiment with ONNX.
- The dataset came from Thiago Martins: https://github.com/thigm85
Training data can be obtained as follows:
import pandas as pd
training_data = pd.read_csv("https://thigm85.github.io/data/cord19/cord19-query-title-label.csv")
training_data.head()
Please do not use this for any clinical/applied purpose. It is a toy app only.
Model info
Created: 2023-10-08
Updated: 2023-10-08
View on Hugging Face