clearml-docs/docs/getting_started/mlops/mlops_second_steps.md
2023-01-12 16:57:08 +02:00

4.2 KiB

title
Next Steps

Once Tasks are defined and in the ClearML system, they can be chained together to create Pipelines. Pipelines provide users with a greater level of abstraction and automation, with Tasks running one after the other.

Tasks can interface with other Tasks in the pipeline and leverage other Tasks' work products.

The sections below describe the following scenarios:

  • Dataset creation
  • Data processing and consumption
  • Pipeline building

Building Tasks

Dataset Creation

Let's assume we have some code that extracts data from a production database into a local folder. Our goal is to create an immutable copy of the data to be used by further steps:

clearml-data create --project data --name dataset
clearml-data sync --folder ./from_production 

We could also add a tag latest to the Dataset, marking it as the latest version.

Preprocessing Data

The second step is to preprocess the date. First we need to access it, then we want to modify it, and lastly we want to create a new version of the data.

# create a task for the data processing part
task = Task.init(project_name='data', task_name='create', task_type='data_processing')

# get the v1 dataset
dataset = Dataset.get(dataset_project='data', dataset_name='dataset_v1')

# get a local mutable copy of the dataset
dataset_folder = dataset.get_mutable_local_copy(
    target_folder='work_dataset', 
    overwrite=True
)
# change some files in the `./work_dataset` folder

# create a new version of the dataset with the pickle file
new_dataset = Dataset.create(
    dataset_project='data', dataset_name='dataset_v2', 
    parent_datasets=[dataset], 
    use_current_task=True,  
    # this will make sure we have the creation code and the actual dataset artifacts on the same Task
)
new_dataset.sync_folder(local_path=dataset_folder)
new_dataset.upload()
new_dataset.finalize()
# now let's remove the previous dataset tag
dataset.tags = []
new_dataset.tags = ['latest']

We passed the parents argument when we created v2 of the Dataset, which inherits all the parent's version content. This not only helps trace back dataset changes with full genealogy, but also makes our storage more efficient, since it only stores the changed and / or added files from the parent versions. When we access the Dataset, it automatically merges the files from all parent versions in a fully automatic and transparent process, as if the files were always part of the requested Dataset.

Training

We can now train our model with the latest Dataset we have in the system. We will do that by getting the instance of the Dataset based on the latest tag (if by any chance we have two Datasets with the same tag we will get the newest). Once we have the dataset we can request a local copy of the data. All local copy requests are cached, which means that if we are accessing the same dataset multiple times we will not have any unnecessary downloads.

# create a task for the model training
task = Task.init(project_name='data', task_name='ingest', task_type='training')

# get the latest dataset with the tag `latest`
dataset = Dataset.get(dataset_tags='latest')

# get a cached copy of the Dataset files 
dataset_folder = dataset.get_local_copy()

# train our model here

Building the Pipeline

Now that we have the data creation step, and the data training step, let's create a pipeline that when executed, will first run the first and then run the second. It is important to remember that pipelines are Tasks by themselves and can also be automated by other pipelines (i.e. pipelines of pipelines).

pipe = PipelineController(
    project='data', 
    name='pipeline demo',
    version="1.0"
)

pipe.add_step(
    name='step 1 data',
    base_project_name='data', 
    base_task_name='create'  
)
pipe.add_step(
    name='step 2 train', 
    parents=['step 1 data', ],
    base_project_name='data', 
    base_task_name='ingest'
)

We could also pass the parameters from one step to the other (for example Task.id). In addition to pipelines made up of Task steps, ClearML also supports pipelines consisting of function steps. See more in the full pipeline documentation here.