From b5bd14edbe102de2828571c0aef5553541157178 Mon Sep 17 00:00:00 2001
From: Valeh Valiollah Pour Amiri <4193454+watiss@users.noreply.github.com>
Date: Sun, 23 Jul 2023 10:23:44 -0700
Subject: [PATCH] wip

---
 scvi/train/_trainingplans.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scvi/train/_trainingplans.py b/scvi/train/_trainingplans.py
index 84573ddf65..0886bf3802 100644
--- a/scvi/train/_trainingplans.py
+++ b/scvi/train/_trainingplans.py
@@ -733,7 +733,7 @@ def __init__(
 
     def initialize_metrics(self, n_classes: int):
         """Initialize metrics."""
-        kwargs = {"task": "multiclass", "num_classes": n_classes}
+        kwargs = {"task": "multiclass", "num_classes": n_classes, "top_k": 1}
         self.train_accuracy = Accuracy(**kwargs)
         self.train_f1 = F1Score(**kwargs)
         self.val_accuracy = Accuracy(**kwargs)