Added file for SolveigModel

Author: sot176

utils/metrics/__init__.py

@@ -1,4 +1,5 @@
-__all__ = ["EntropyPrediction", "Recall"]
+__all__ = ["EntropyPrediction", "Recall", "F1"]
 
 from .EntropyPred import EntropyPrediction
 from .recall import Recall
+from .F1 import F1Score

utils/models/__init__.py

@@ -1,4 +1,5 @@
-__all__ = ["MagnusModel", "ChristianModel"]
+__all__ = ["MagnusModel", "ChristianModel", "SolveigModel"]
 
 from .christian_model import ChristianModel
 from .magnus_model import MagnusModel
+from .solveig_model import SolveigModel

utils/models/solveig_model.py

@@ -0,0 +1,11 @@
+import pytest
+import torch
+import torch.nn as nn
+
+
+class SolveigModel(nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    def forward(self, x):
+        return