fix: rm hcn model and trackers

deeppavlov/core/common/registry.json

@@ -1,7 +1,6 @@
 {
   "api_requester": "deeppavlov.models.api_requester.api_requester:ApiRequester",
   "api_router": "deeppavlov.models.api_requester.api_router:ApiRouter",
-  "babi_reader": "deeppavlov.dataset_readers.babi_reader:BabiDatasetReader",
   "basic_classification_iterator": "deeppavlov.dataset_iterators.basic_classification_iterator:BasicClassificationDatasetIterator",
   "basic_classification_reader": "deeppavlov.dataset_readers.basic_classification_reader:BasicClassificationDatasetReader",
   "bow": "deeppavlov.models.embedders.bow_embedder:BoWEmbedder",
@@ -11,7 +10,7 @@
   "conll2003_reader": "deeppavlov.dataset_readers.conll2003_reader:Conll2003DatasetReader",
   "data_fitting_iterator": "deeppavlov.core.data.data_fitting_iterator:DataFittingIterator",
   "data_learning_iterator": "deeppavlov.core.data.data_learning_iterator:DataLearningIterator",
-  "default_tracker": "deeppavlov.models.trackers.default_tracker:DefaultTracker",
+  "default_tracker": "deeppavlov.models.go_bot.tracker:DefaultTracker",
   "default_vocab": "deeppavlov.core.data.vocab:DefaultVocabulary",
   "dialog_db_result_iterator": "deeppavlov.dataset_iterators.dialog_iterator:DialogDBResultDatasetIterator",
   "dialog_iterator": "deeppavlov.dataset_iterators.dialog_iterator:DialogDatasetIterator",
@@ -31,8 +30,6 @@
   "go_bot": "deeppavlov.models.go_bot.bot:GoalOrientedBot",
   "go_bot_rnn": "deeppavlov.models.go_bot.network:GoalOrientedBotNetwork",
   "hashing_tfidf_vectorizer": "deeppavlov.models.vectorizers.hashing_tfidf_vectorizer:HashingTfIdfVectorizer",
-  "hcn_at": "deeppavlov.models.trackers.hcn_at:ActionTracker",
-  "hcn_et": "deeppavlov.models.trackers.hcn_et:EntityTracker",
   "insurance_reader": "deeppavlov.dataset_readers.insurance_reader:InsuranceReader",
   "kenlm_elector": "deeppavlov.models.spelling_correction.electors.kenlm_elector:KenlmElector",
   "keras_classification_model": "deeppavlov.models.classifiers.keras_classification_model:KerasClassificationModel",
@@ -90,4 +87,4 @@
   "typos_wikipedia_reader": "deeppavlov.dataset_readers.typos_reader:TyposWikipedia",
   "wiki_sqlite_vocab": "deeppavlov.vocabs.wiki_sqlite:WikiSQLiteVocab",
   "wikitionary_100K_vocab": "deeppavlov.vocabs.typos:Wiki100KDictionary"
-}
+}

deeppavlov/models/go_bot/tracker.py

@@ -20,6 +20,54 @@
 from deeppavlov.core.common.registry import register
 
 
+@register('default_tracker')
+class DefaultTracker(Component):
+
+    def __init__(self, slot_names):
+        self.slot_names = list(slot_names)
+        self.reset_state()
+
+    @property
+    def state_size(self):
+        return len(self.slot_names)
+
+    @property
+    def num_features(self):
+        return self.state_size
+
+    def reset_state(self):
+        self.history = []
+        self.curr_feats = np.zeros(self.num_features, dtype=np.float32)
+
+    def update_state(self, slots):
+        def _filter(slots):
+            return filter(lambda s: s[0] in self.slot_names, slots)
+        if type(slots) == list:
+            self.history.extend(_filter(slots))
+        elif type(slots) == dict:
+            for slot, value in _filter(slots.items()):
+                self.history.append((slot, value))
+        self.curr_feats = self._binary_features()
+        return self
+
+    def get_state(self):
+        lasts = {}
+        for slot, value in self.history:
+            lasts[slot] = value
+        return lasts
+
+    def _binary_features(self):
+        feats = np.zeros(self.state_size, dtype=np.float32)
+        lasts = self.get_state()
+        for i, slot in enumerate(self.slot_names):
+            if slot in lasts:
+                feats[i] = 1.
+        return feats
+
+    def __call__(self):
+        return self.curr_feats
+
+
 @register('featurized_tracker')
 class FeaturizedTracker(Component):