Skip to content

Commit 6a1717f

Browse files
committed
shrink code size of sample
1 parent d6cd285 commit 6a1717f

File tree

1 file changed

+7
-17
lines changed

1 file changed

+7
-17
lines changed

README.md

Lines changed: 7 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -56,25 +56,15 @@ Consider using TopicNet if:
5656
And here is sample code of the TopicNet baseline experiment:
5757

5858
```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
7161

7262
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)],
7868
)
7969
experiment, dataset = (
8070
build_experiment_environment_from_yaml_config(

0 commit comments

Comments
 (0)