|
@@ -15,7 +15,7 @@ parser.add_argument('--target_index', type=int, default=6, |
|
|
help='Index of the ground truth (for evaluation)') |
|
|
help='Index of the ground truth (for evaluation)') |
|
|
parser.add_argument('--metric',type=str, default='F1_MACRO', |
|
|
parser.add_argument('--metric',type=str, default='F1_MACRO', |
|
|
help='Evaluation Metric (F1, F1_MACRO)') |
|
|
help='Evaluation Metric (F1, F1_MACRO)') |
|
|
parser.add_argument('--pipeline_path', default=os.path.join(this_path, './example_pipelines/abod_pipeline.json'), |
|
|
|
|
|
|
|
|
parser.add_argument('--pipeline_path', default=os.path.join(this_path, './example_pipelines/autoencoder_pipeline.json'), |
|
|
help='Input the path of the pre-built pipeline description') |
|
|
help='Input the path of the pre-built pipeline description') |
|
|
# parser.add_argument('--pipeline_path', default=os.path.join(this_path, '../tods/resources/default_pipeline.json'), |
|
|
# parser.add_argument('--pipeline_path', default=os.path.join(this_path, '../tods/resources/default_pipeline.json'), |
|
|
# help='Input the path of the pre-built pipeline description') |
|
|
# help='Input the path of the pre-built pipeline description') |
|
@@ -33,9 +33,6 @@ dataset = generate_dataset(df, target_index) |
|
|
|
|
|
|
|
|
# Load the default pipeline |
|
|
# Load the default pipeline |
|
|
pipeline = load_pipeline(pipeline_path) |
|
|
pipeline = load_pipeline(pipeline_path) |
|
|
print(dir(pipeline)) |
|
|
|
|
|
print(pipeline.steps) |
|
|
|
|
|
print(dir(dataset)) |
|
|
|
|
|
|
|
|
|
|
|
# Run the pipeline |
|
|
# Run the pipeline |
|
|
pipeline_result = evaluate_pipeline(dataset, pipeline, metric) |
|
|
pipeline_result = evaluate_pipeline(dataset, pipeline, metric) |
|
|