kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-11-11 22:53:55 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: fix adding user and anime threshold argument

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2023-05-29 23:30:20 +02:00
parent 403e7bfec5
commit c15e9d6618
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
midterm/code/main.py
View File

@ -3,6 +3,7 @@ Code for preprocessing data and creating model that predicts and
recomends anime based on another anime entered by user recomends anime based on another anime entered by user
""" """
import pandas as pd import pandas as pd
import numpy as np import numpy as np
import argparse import argparse
@ -220,9 +221,9 @@ def handle_arguments():
parser.add_argument('--neighbors', '-n', help='Specify number of nearest neighbors', parser.add_argument('--neighbors', '-n', help='Specify number of nearest neighbors',
required=False, default=5) required=False, default=5)
parser.add_argument('--user_threshold', '-ut', help='Specify minimal number of votes required for user to be included in the data, set to -1 for no threshold', parser.add_argument('--user_threshold', '-ut', help='Specify minimal number of votes required for user to be included in the data, set to -1 for no threshold',
required=False, default=500) required=False, type=int, default=500)
parser.add_argument('--anime_threshold', '-at', help='Specify minimal number of votes required for anime to be included in the data, set to -1 for no threshold', parser.add_argument('--anime_threshold', '-at', help='Specify minimal number of votes required for anime to be included in the data, set to -1 for no threshold',
required=False, default=200) required=False, type=int, default=200)
# Parse the command-line arguments # Parse the command-line arguments
args = parser.parse_args() args = parser.parse_args()