Veucci commited on
Commit
ef131d2
1 Parent(s): 033a08e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -20,7 +20,7 @@ widget:
20
 
21
  ## Description
22
 
23
- The model was trained using the BERT language model on a 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.
24
  For training and test codes check out [Github page](https://github.com/Veucci/lyrics-to-genre-lite).
25
 
26
  ## Dataset
 
20
 
21
  ## Description
22
 
23
+ The model was trained using the BERT language model on my [song lyrics dataset](https://huggingface.co/datasets/Veucci/lyrics_3genre) 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.
24
  For training and test codes check out [Github page](https://github.com/Veucci/lyrics-to-genre-lite).
25
 
26
  ## Dataset