Update prediction_multilabel.py (#6)

- Update prediction_multilabel.py (8bb6ca89c2a7aa3e550e485bed41c01e3087ec73)


Co-authored-by: Hamid Bekamiri <[email protected]>

prediction_multilabel.py

@@ -10,15 +10,15 @@ from sentence_transformers import util
 torch.manual_seed(1)
 
 # Load datasets
-df_inmemory = pd.read_csv('df_360k_41lables_05012023.csv')
-df_paragraph = pd.read_csv('german_plc_all_paragraphs_unnested_only.csv')
+df_inmemory = pd.read_csv('embeddings_labeled.csv')
+df_paragraph = pd.read_csv('embeddings_prediction.csv')
 
 # Load stored embeddings
-with open('embeddings_paragraphs_07012023.pkl', "rb") as f:
+with open('embeddings_prediction.pkl.pkl', "rb") as f:
     stored_data = pickle.load(f)
     pred_embeddings = stored_data['parg_embeddings']
 
-with open('embeddings_sentences_360k_09012023.pkl', "rb") as f:
+with open('embeddings_labeled.pkl.pkl', "rb") as f:
     stored_data = pickle.load(f)
     embeddings = stored_data['sent_embeddings']