From 3ce66a664bf40e1ed0acb2c0e5a21419b17adede Mon Sep 17 00:00:00 2001 From: Hetvi Shastri Date: Wed, 21 Jul 2021 04:54:41 +0530 Subject: [PATCH] modified init.py for new networks --- nilmtk_contrib/disaggregate/__init__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/nilmtk_contrib/disaggregate/__init__.py b/nilmtk_contrib/disaggregate/__init__.py index 553830c..9d560df 100644 --- a/nilmtk_contrib/disaggregate/__init__.py +++ b/nilmtk_contrib/disaggregate/__init__.py @@ -6,4 +6,9 @@ from .seq2point import Seq2Point from .seq2seq import Seq2Seq from .WindowGRU import WindowGRU -from .rnn import RNN \ No newline at end of file +from .rnn import RNN +from .rnn_attention import RNN_attention +from .rnn_attention_classification import RNN_attention_classification +from .resnet import ResNet +from .resnet_classification import ResNet_classification +from .bert import BERT \ No newline at end of file