mirror of
https://github.com/lukefleed/imdb-graph.git
synced 2026-10-09 14:41:42 +00:00
final filtering refactoring
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user