Oracle-Baseline
Implent of Oracle Base line and Lea-3 Baseline
Oracle
Oracle : This model is used to obtain an oracle with a greedy algorithm similar to Nallapati et al. (2017)
, which treats the utterances that maximize the ROUGE-2 as a summary.
python train_oracle.py
Please remember to change file path in this code , qwq. You can change the number of the oracle you want by changing summary_size in greedy_selection.
Lead-3
Lead-3: This model simply selects the first three sentences in a document as summary.
python train_lead.py
Please remember to change file path in this code , qwq. You can change the number of the oracle you want by changing lead_size in lead_selection.
Calculate Rouge
You can use cal_rouge.py to calculate rouge scores here.
Requirements
rouge==1.0.1
nltk==3.6.5
regex == 2021.8.3
