Spaces:
Sleeping
Sleeping
artifacts_root: artifacts | |
data_ingestion: | |
dataset_name: "alexfabbri/multi_news" | |
arrow_dataset_dir: artifacts/data | |
data_validation: | |
root_dir: artifacts/data_validation | |
status_file: artifacts/data_validation/status.txt | |
all_required_folders: ["train", "test", "validation", "dataset_dict.json"] | |
data_transformation: | |
root_dir: artifacts/data_transformation | |
data_path: artifacts/data | |
tokenizer_name: google/pegasus-cnn_dailymail | |
model_trainer: | |
root_dir: artifacts/model_trainer | |
data_path: artifacts/data_transformation/dataset | |
model_ckpt: google/pegasus-cnn_dailymail | |
model_path: artifacts/model_trainer/pegasus-samsum-model | |
tokenizer_path: artifacts/model_trainer/tokenizer | |
model_evaluation: | |
root_dir: artifacts/model_evaluation | |
data_path: artifacts/data_transformation/dataset | |
model_path: artifacts/model_trainer/pegasus-samsum-model | |
tokenizer_path: artifacts/model_trainer/tokenizer | |
metric_file_name: artifacts/model_evaluation/metrics.csv | |
hub_model_name: pavithra-devi/pegasus-multi-news | |