AIR

Original dataset:

Word to vector model:

Dependency manager:

Abba song retrieval. Model outputs whether a document is relevant for the given query. Baseline created with cosine similarity.

Name	Name	Last commit message	Last commit date
Latest commit History 27 Commits 27 Commits
.idea	.idea
plots	plots
.gitignore	.gitignore
README.md	README.md
baseline.py	baseline.py
cos_test_f1.csv	cos_test_f1.csv
cos_test_loss.csv	cos_test_loss.csv
cos_train_f1.csv	cos_train_f1.csv
cos_train_loss.csv	cos_train_loss.csv
main.py	main.py
model.py	model.py
plots.py	plots.py
poetry.lock	poetry.lock
preprocessing.py	preprocessing.py
pyproject.toml	pyproject.toml
spotify_abba_songdata.csv	spotify_abba_songdata.csv
test_f1.csv	test_f1.csv
test_loss.csv	test_loss.csv
train_f1.csv	train_f1.csv
train_loss.csv	train_loss.csv

Provide feedback