From 68480992b77e571523f2f854910383c7523399d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=A9=E9=82=91?= Date: Wed, 9 Oct 2024 20:56:51 +0800 Subject: [PATCH] fix seq_group_name attr --- tzrec/models/tdm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tzrec/models/tdm.py b/tzrec/models/tdm.py index 5d25988..8707fd8 100644 --- a/tzrec/models/tdm.py +++ b/tzrec/models/tdm.py @@ -43,6 +43,7 @@ def __init__( ) non_seq_fea_dim = 0 + self.seq_group_name = "" self.non_seq_group_name = [] query_emb_dim = 0 for feature_group in model_config.feature_groups: