From e9cecd326586908514d32ccc4b00ac098fa82faf Mon Sep 17 00:00:00 2001
From: Frank Sauerburger <frank@sauerburger.com>
Date: Sun, 28 Aug 2022 13:19:41 +0200
Subject: [PATCH] Remove duplicated function
---
movies.py | 9 ---------
1 file changed, 9 deletions(-)
diff --git a/movies.py b/movies.py
index 088e27d..dc8f8e5 100644
--- a/movies.py
+++ b/movies.py
@@ -350,15 +350,6 @@ def log_preduction_model(model):
pip_requirements="requirements.txt")
-def build_prediction_model(model, movies):
- """Build an index to get movie suggestions"""
- index = tfrs.layers.factorized_top_k.BruteForce(model.query_model)
- index.index_from_dataset(tf.data.Dataset.zip(
- (movies.batch(100), movies.batch(100).map(model.cand_model))
- ))
- return index
-
-
def train(args):
"""Load the dataset, train and evaluate the model"""
mlflow.set_experiment(args.mlflow_experiment)
--
GitLab