@@ -56,25 +56,15 @@ Consider using TopicNet if:
56
56
And here is sample code of the TopicNet baseline experiment:
57
57
58
58
``` python
59
- from topicnet.cooking_machine.config_parser import (
60
- build_experiment_environment_from_yaml_config
61
- )
62
-
63
- from topicnet.cooking_machine.recipes import (
64
- ARTM_baseline as config_string
65
- )
66
-
67
- dataset_path = ' /data/datasets/NIPS/dataset.csv'
68
-
69
- specific_topics = [f ' spc_topic_ { i} ' for i in range (19 )]
70
- background_topics = [f ' bcg_topic_ { i} ' for i in range ( 1 )]
59
+ from topicnet.cooking_machine.config_parser import build_experiment_environment_from_yaml_config
60
+ from topicnet.cooking_machine.recipes import ARTM_baseline as config_string
71
61
72
62
config_string = config_string.format(
73
- dataset_path = dataset_path ,
74
- modality_list = [' @word' ],
75
- main_modality = ' @word' ,
76
- specific_topics = specific_topics ,
77
- background_topics = background_topics ,
63
+ dataset_path = ' /data/datasets/NIPS/dataset.csv ' ,
64
+ modality_list = [' @word' ],
65
+ main_modality = ' @word' ,
66
+ specific_topics = [ f ' spc_topic_ { i } ' for i in range ( 19 )] ,
67
+ background_topics = [ f ' bcg_topic_ { i } ' for i in range ( 1 )] ,
78
68
)
79
69
experiment, dataset = (
80
70
build_experiment_environment_from_yaml_config(
0 commit comments