fix init avitm and nfm models
@@ -11,7 +11,7 @@ class AVITM(Abstract_Model):
dropout=0.2, learn_priors=True, batch_size=64, lr=2e-3, momentum=0.99,
solver='adam', num_epochs=100, reduce_on_plateau=False, prior_mean=0.0,
prior_variance=None, num_layers=2, num_neurons=100, use_partitions=True):
- super(Abstract_Model).__init__()
+ super().__init__()
self.hyperparameters['num_topics'] = num_topics
self.hyperparameters['model_type'] = model_type
self.hyperparameters['activation'] = activation
