From b28dc138acbe0eeec99b719c34c2a79880fd518a Mon Sep 17 00:00:00 2001 From: Luca Lombardo Date: Tue, 19 Apr 2022 10:03:54 +0200 Subject: [PATCH] final filtering refactoring --- filters/actors_graph_filter.py | 2 +- filters/movie_graph_filter.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/filters/actors_graph_filter.py b/filters/actors_graph_filter.py index 753e6c8..199d0f1 100755 --- a/filters/actors_graph_filter.py +++ b/filters/actors_graph_filter.py @@ -7,7 +7,7 @@ import numpy as np import os import csv -# MIN_MOVIES = 5 # Only keep relations for actors that have made more than this many movies +# MIN_MOVIES = 5 # Deprecated, now user gives this value as input parser = argparse.ArgumentParser() parser.add_argument("--min-movies", type=int, required=True) diff --git a/filters/movie_graph_filter.py b/filters/movie_graph_filter.py index a2d519a..65bbb25 100755 --- a/filters/movie_graph_filter.py +++ b/filters/movie_graph_filter.py @@ -9,6 +9,8 @@ import csv #-----------------DOWNLOAD .GZ FILES FROM IMDB DATABASE-----------------# +# User gives as input the minimun number of votes that a movie needs to have on the IMDb database to be considered + parser = argparse.ArgumentParser() parser.add_argument("--votes", type=int, required=True) args = parser.parse_args()