search

hanjanghoon / BERT_FP

Fine-grained Post-training for Improving Retrieval-based Dialogue Systems - NAACL 2021
97 stars 19 forks source link

Using on custom data #5

Closed vt404v2 closed 2 years ago

vt404v2 commented 2 years ago

Hello! Can you please give me an advise how to use this model on my custom data on the inference? I want to give to model something like list of dialog history and list of hypotheses and receive probability distribution for hypotheses. Are there any easy way to do that?

hanjanghoon commented 2 years ago

please refers Data_preprocessing.py