final filtering refactoring

main
Luca Lombardo 3 years ago
parent 0b2b670b24
commit b28dc138ac

@ -7,7 +7,7 @@ import numpy as np
import os import os
import csv 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 = argparse.ArgumentParser()
parser.add_argument("--min-movies", type=int, required=True) parser.add_argument("--min-movies", type=int, required=True)

@ -9,6 +9,8 @@ import csv
#-----------------DOWNLOAD .GZ FILES FROM IMDB DATABASE-----------------# #-----------------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 = argparse.ArgumentParser()
parser.add_argument("--votes", type=int, required=True) parser.add_argument("--votes", type=int, required=True)
args = parser.parse_args() args = parser.parse_args()

Loading…
Cancel
Save