Edit model card

Entailment Detection by Fine-tuning BERT


  • The model in this repository is fine-tuned on Google's encoder-decoder transformer-based model BERT.
  • New York University's Multi-NLI dataset is used for fine-tuning.
  • Accuracy achieved: ~74%

    image/png

  • Notebook used for fine-tuning: here
  • N.B.: Due to computational resource constraints, only 11K samples are used for fine-tuning. There is room for accuracy improvement if a model is trained on all the 390K samples available in the dataset.
    Downloads last month
    0
    Inference Examples
    Inference API (serverless) is not available, repository is disabled.

    Model tree for ArghaKamalSamanta/ema_task_entailment

    Adapter
    this model

    Dataset used to train ArghaKamalSamanta/ema_task_entailment