Skip to content

jjxxmiin/neural-collaborative-filtering-mlops

Repository files navigation

Neural Collaborative Filtering

  • OS: Window
  • Dataset: MovieLens
  • Framework: Pytorch

Tools for MLOPS

  • Model Versioning: DVC
  • Experiments: MLFlow
  • Config: Hydra
  • Deployment: BentoML

MLFlow

mlflow run . --env-manager=local --entry-point gmf --experiment-name gmf
mlflow run . --env-manager=local --entry-point mlp --experiment-name mlp
mlflow run . --env-manager=local --entry-point neumf --experiment-name neumf
mlflow ui
  • model training
  • experiments
  • logging

DVC

dvc init 
dvc remote add -d checkpoints {LOCAL, S3, etc.}
dvc add
dvc push
dvc pull

git commit -m "commit"
git tag -a "v1.0" -m "model, data v1.0"
  • model & data versioning

Reference

About

neural collaborative filtering using mlops

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages