film_mlmodule / pipelines /training_pipeline.py
root
first hf commit
6defa3d
raw
history blame
No virus
449 Bytes
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)