Downloads · 30 days
9
1% of all-time downloads
lyric-to-genre
Veucci/lyric-to-genre
lyric-to-genre is a text classification model from Veucci. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as cc-by-nc-4.0.
The model was trained using the BERT language model on my song lyrics dataset to predict the genre of a given song based on its lyrics. This repository houses the machine learning model, which is capable of making pre…
All-time downloads
1.5K
Public
Parameters
109M
876 MB on disk
Likes
4
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 50%
Parameter types
How the weights are stored.
F32109M · 100%
Try a prompt
Datasets
Spaces
Model card
From the Hugging Face model README
Lyrics Genre Classification Model
Description
The model was trained using the BERT language model on my song lyrics dataset to predict the genre of a given song based on its lyrics. This repository houses the machine learning model, which is capable of making predictions in three distinct genres: Pop, Rock, and Hip-Hop. For training and test codes check out Github page.
Dataset
The model was trained on a diverse and labeled dataset of song lyrics, which contained approximately 3000 rows. The dataset was carefully curated to include songs from a wide range of artists and genres, ensuring a comprehensive representation of Pop, Rock, and Hip-Hop music. DATASET
Quick Start
from transformers import pipeline
classifier = pipeline("text-classification", model="Veucci/lyrics-to-genre")
result = classifier("When I'm away from you, I'm happier than ever Wish I could explain it better I wish it wasn't true")
print(result)
License
This dataset is released under the Creative Commons Attribution-NonCommercial license. This means that you are not allowed to use the dataset for commercial purposes. For detailed information about the license, please refer to the LICENSE file.
Contact
If you have any questions, suggestions, or concerns regarding this dataset, please feel free to reach out to email at [email protected]. I hope this model helps in your genre classification tasks and inspires further exploration of song lyrics analysis!
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- cc-by-nc-4.0
- Languages
- en
- Created
- Jul 2, 2023
- Updated
- Aug 10, 2023