ia-nechaev
/

sbic-method2

Text Classification

Model card Files Files and versions

Update prediction_multilabel.py

#12

by HamidBekam - opened Mar 19, 2025

base: refs/heads/main

←

from: refs/pr/12

Discussion Files changed

Files changed (1) hide show

prediction_multilabel.py +2 -2

prediction_multilabel.py CHANGED Viewed

@@ -10,8 +10,8 @@ from sentence_transformers import util
 torch.manual_seed(1)
 # Load datasets
-df_inmemory = pd.read_csv('embeddings_labeled.csv')
-df_paragraph = pd.read_csv('embeddings_prediction.csv')
 # Load stored embeddings
 with open('embeddings_prediction.pkl.pkl', "rb") as f:

 torch.manual_seed(1)
 # Load datasets
+df_inmemory = pd.read_csv('labeled.csv')
+df_paragraph = pd.read_csv('prediction.csv')
 # Load stored embeddings
 with open('embeddings_prediction.pkl.pkl', "rb") as f: