File size: 449 Bytes
6defa3d
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from zenml import pipeline

from steps.data_retrieving import retrieve_data
from steps.data_ingesting import ingest_data
from steps.data_preprocessing import preprocess_data
from steps.model_train import train_model
from steps.evaluation import evaluate_model

@pipeline(enable_cache=False)
def train_pipeline(data_path: str):
    df_movie, df_user, df_rating = retrieve_data()
    df_total = preprocess_data(df_movie, df_user, df_rating)