From d98395930438cc4292d6b2ae0c4d5e375c599ea1 Mon Sep 17 00:00:00 2001
From: Daniel Valdenegro Ibarra <dhvalden@gmail.com>
Date: Wed, 30 Mar 2022 16:20:19 +0100
Subject: [PATCH] Create example_predict.yml

---
 example_predict.yml | 13 +++++++++++++
 1 file changed, 13 insertions(+)
 create mode 100644 example_predict.yml

diff --git a/example_predict.yml b/example_predict.yml
new file mode 100644
index 0000000..27f377d
--- /dev/null
+++ b/example_predict.yml
@@ -0,0 +1,13 @@
+HF:
+  model_arch: distilbert
+  model_name: distilbert-base-uncased
+CONFIG:
+  data_path: data/data_new.csv
+  text_col: 0
+  out_path: data/preds.csv
+  batch_size: 32
+  state_dict_path: models/model1_distilBERT_epoch_5.model
+  labels:
+        label1: 0
+        label2: 1
+        label3: 2