Skip to content

aaaa760/Semantic-Abstract-Generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Semantic Abstract Generator

This repository contains the source code for a study on generating abstracts for given texts using a fine-tuned T5-base transformer model.

Dataset

The dataset used in this study can be found at Dataset.

The subset of the dataset i used for training the model can be found at Dataset.

Requirements

To run the code in this repository, you will need to have Python 3.6 or later installed, along with the following libraries:

  • transformers
  • torch
  • numpy
  • pandas

Results

The model achieved a precision of 0.5, a recall of 0.6, and an f1 score of 0.6 on the test dataset.

Average semantic similarity: 0.268

Average cosine similarity: 0.451

The Report for the particular project can be found at Report

License

This project is licensed under the MIT License. See the LICENSE file for more information.

About

A abstract generatior using T5 transformer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published