Skip to content

Commit

Permalink
feat(corprep): add run configurations for absa_agent_predict, filter_…
Browse files Browse the repository at this point in the history
…dataset and load_raw_dataset
  • Loading branch information
entelecheia committed Jul 26, 2023
1 parent f9dfb5b commit 7290fca
Show file tree
Hide file tree
Showing 6 changed files with 31 additions and 20 deletions.
11 changes: 1 addition & 10 deletions src/corprep/conf/pipe/absa_agent_predict.yaml
Original file line number Diff line number Diff line change
@@ -1,15 +1,6 @@
defaults:
- __general_external_funcs__
- /run: absa_agent_predict

run: corprep.absa.agent.predict
run_with:
tasks: [AE]
absa_config_name: default
text_col: bodyText
batch_size: 10
num_workers: 1
remove_columns:
load_from_cache_file: true
verbose: ${..verbose}
use_pipe_obj: true
return_pipe_obj: false
7 changes: 1 addition & 6 deletions src/corprep/conf/pipe/dataset_filter.yaml
Original file line number Diff line number Diff line change
@@ -1,12 +1,7 @@
defaults:
- __general_external_funcs__
- /run: filter_dataset

run: corprep.datasets.preprocessing.filter.filter_dataset
run_with:
num_proc: 1
batched: true
text_col: bodyText
verbose: ${..verbose}
use_pipe_obj: true
return_pipe_obj: false
# 은는이가도과와에의등
Expand Down
6 changes: 2 additions & 4 deletions src/corprep/conf/pipe/dataset_load_raw.yaml
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
defaults:
- __general_external_funcs__
- /run: load_raw_dataset

run: corprep.datasets.io.load_raw_dataset
run_with:
raw_dataset_dir:
run:
path: json
verbose: false
use_pipe_obj: false
12 changes: 12 additions & 0 deletions src/corprep/conf/run/absa_agent_predict.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
defaults:
- __init__

_target_: corprep.absa.agent.predict
tasks: [AE]
absa_config_name: default
text_col: bodyText
batch_size: 10
num_workers: 1
remove_columns:
load_from_cache_file: true
verbose: ${oc.select:..verbose, false}
8 changes: 8 additions & 0 deletions src/corprep/conf/run/filter_dataset.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
defaults:
- __init__

_target_: corprep.datasets.preprocessing.filter.filter_dataset
num_proc: 1
batched: true
text_col: bodyText
verbose: ${oc.select:..verbose, false}
7 changes: 7 additions & 0 deletions src/corprep/conf/run/load_raw_dataset.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
defaults:
- __init__

_target_: corprep.datasets.io.load_raw_dataset
raw_dataset_dir:
path: json
verbose: ${oc.select:..verbose, false}

0 comments on commit 7290fca

Please sign in to comment.