Update optimization_demo_AVITM.ipynb
"from dataset.dataset import Dataset\n",
"from evaluation_metrics.diversity_metrics import Topic_diversity\n",
"from evaluation_metrics.topic_significance_metrics import KL_uniform\n",
- "#from skopt import gp_minimize, forest_minimize, dummy_minimize\n",
- "#from optimization.optimizer import Optimizer\n",
- "#from skopt.space.space import Real, Integer\n",
+ "from skopt import gp_minimize, forest_minimize, dummy_minimize\n",
+ "from optimization.optimizer import Optimizer\n",
+ "from skopt.space.space import Real, Integer\n",
"import multiprocessing as mp\n",
"from models.TorchAvitm import TorchAvitm\n",
"from gensim.models import Word2Vec"
