reduced requirements

environment.yml

@@ -8,15 +8,11 @@ dependencies:
   - python=3.10
   - pytorch
   - torchaudio
-  - torchvision
   - librosa
   - numpy
   - pandas
-  - seaborn
-  - matplotlib
   - bs4
   - requests
-  - bidict
   - tqdm
   - pytorch-lightning
   - rich
@@ -25,9 +21,3 @@ dependencies:
   - transformers
   - accelerate
   - pytest
-
-  - pip:
-      - evaluate
-      - wakepy
-      - soundfile
-      - youtube_dl

models/utils.py

@@ -1,13 +1,9 @@
 import torch.nn as nn
 import torch
 import numpy as np
-import evaluate
 from sklearn.metrics import precision_score, recall_score, f1_score, accuracy_score
 
 
-accuracy = evaluate.load("accuracy")
-
-
 class LabelWeightedBCELoss(nn.Module):
     """
     Binary Cross Entropy loss that assumes each float in the final dimension is a binary probability distribution.
@@ -86,4 +82,4 @@ def get_id_label_mapping(labels: list[str]) -> tuple[dict, dict]:
 
 def compute_hf_metrics(eval_pred):
     predictions = np.argmax(eval_pred.predictions, axis=1)
-    return accuracy.compute(predictions=predictions, references=eval_pred.label_ids)
+    return accuracy_score(y_true=eval_pred.label_ids, y_pred=predictions)

requirements.txt

@@ -1,5 +1,4 @@
 torch
-torchvision
 torchaudio
 pytorch-lightning
 numpy