Downloads · 30 days
22
0% of all-time downloads
HannaAbiAkl/flan-t5-small-wordnet
flan-t5-small-wordnet is a text generation model from HannaAbiAkl. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
This model is a fine-tuned version of flan-t5-small on the WordNet dataset.
Downloads · 30 days
22
0% of all-time downloads
All-time downloads
10.7K
Public
Parameters
77M
309 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors308 MB · 99%
From the Hugging Face model README
This model is a fine-tuned version of flan-t5-small on the WordNet dataset.
The model is trained to classify terms into one of four term types: noun, verb, adjective or adverb. The types themselves are learned and then generated by the model with no more than one type associated with a specific term.
The model also works well as part of a Retrieval-and-Generation (RAG) pipeline by leveraging an external knowledge source, specifically Wordnet Semantic Primes.
This model is intended to be used to generate a type (class) for an input term.
The training and evaluation data can be found here.
The train size is 40559.
The test size is 9470.
Here's an example of the model capabilities:
input:
output:
input:
output:
input:
output:
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.1725 | 1.0 | 1000 | 0.0640 |
| 0.1250 | 2.0 | 2000 | 0.0535 |
| 0.1040 | 3.0 | 3000 | 0.0469 |
| 0.0917 | 4.0 | 4000 | 0.0421 |
| 0.0830 | 5.0 | 5000 | 0.0384 |
@misc{akl2024dstillms4ol2024task,
title={DSTI at LLMs4OL 2024 Task A: Intrinsic versus extrinsic knowledge for type classification},
author={Hanna Abi Akl},
year={2024},
eprint={2408.14236},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2408.14236},
}