File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed
autointent/generation/utterances/evolution Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change 1313from datasets import concatenate_datasets
1414
1515from autointent import Dataset , Pipeline
16- from autointent .configs import EmbedderConfig
1716from autointent .custom_types import Split
1817from autointent .generation .utterances import UtteranceEvolver
1918from autointent .generation .utterances .generator import Generator
@@ -83,8 +82,6 @@ def augment(
8382
8483 Note that for now it supports only single-label datasets.
8584 """
86- emb_config = EmbedderConfig (batch_size = 16 , device = "cuda" )
87-
8885 best_result = 0
8986 merge_dataset = copy .deepcopy (dataset )
9087
@@ -95,7 +92,6 @@ def augment(
9592 merge_dataset [split_name ] = concatenate_datasets ([merge_dataset [split_name ], new_samples_dataset ])
9693
9794 pipeline_optimizer = Pipeline .from_search_space (self .search_space )
98- pipeline_optimizer .set_config (emb_config )
9995 ctx = pipeline_optimizer .fit (merge_dataset )
10096 results = ctx .optimization_info .dump_evaluation_results ()
10197 decision_metric = results ["metrics" ]["decision" ][0 ]
You can’t perform that action at this time.
0 commit comments