kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-04 20:43:06 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: print only the most recommended anime

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2023-06-11 22:39:34 +02:00
parent 4a66ce8731
commit 5f2a5c8836
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
final/code/main.py
View File

@ -187,7 +187,7 @@ def predict(prediction_model, pivot_table, seed=42, anime="RANDOM", recommendati
query)
if debug:
print("prediction model, distance: ", distance)
for i in range(recommendation_number):
for i in range(0, 1):
if i == 0 and not auto and not debug:
print(f"Recommendations for {chosen_anime_name}:\n")
elif not auto and not debug:
@ -315,12 +315,13 @@ def auto_mode(data_limit=-1, seed=42, anime="RANDOM"):
if possibleMetrics == []:
possibleMetrics = [""]
for metric in possibleMetrics:
print("testing for algorithm, metric: ", algorithm, metric)
for neighbor_amount in neighbor_spread:
print("testing for algorithm, metric, neighbor_amount: ",
algorithm, metric, neighbor_amount)
preprocess_model_predict(starting_rating_data, starting_anime_contact_data,
starting_rows_number, original_pivot_table, seed=seed, anime=anime, neighbors=neighbor_amount, algorithm=algorithm, metric=metric)
if metric != 'precomputed':
print("testing for algorithm, metric: ", algorithm, metric)
for neighbor_amount in neighbor_spread:
print("testing for algorithm, metric, neighbor_amount: ",
algorithm, metric, neighbor_amount)
preprocess_model_predict(starting_rating_data, starting_anime_contact_data,
starting_rows_number, original_pivot_table, seed=seed, anime=anime, neighbors=neighbor_amount, algorithm=algorithm, metric=metric)
def write_test_results(title, result=""):