Merge branch 'main' of https://repos.hackathon.bz.it/2025-summer/team-1
This commit is contained in:
commit
8fc79437e6
2 changed files with 32 additions and 2 deletions
|
@ -2,7 +2,16 @@ from dataclasses import dataclass
|
|||
|
||||
from song import Song
|
||||
|
||||
type UserScoredSong = tuple[Song, int]
|
||||
USER_SCORE_WEIGHT = 0.7
|
||||
GENRE_WEIGHT = 0.1
|
||||
RANDOM_WEIGHT = 0.1
|
||||
RECENT_PENALTY = 0.5
|
||||
RECENT_COUNT = 10
|
||||
|
||||
|
||||
@dataclass
|
||||
class UserScoredSong(Song):
|
||||
user_score: int
|
||||
|
||||
|
||||
@dataclass
|
||||
|
@ -14,5 +23,25 @@ class Room:
|
|||
tags: set[str]
|
||||
creative: bool
|
||||
|
||||
playlist: set[UserScoredSong]
|
||||
songs: dict[str, UserScoredSong]
|
||||
history: list[Song]
|
||||
|
||||
def rank_song(
|
||||
self,
|
||||
song: UserScoredSong,
|
||||
) -> float:
|
||||
rank = 0.0
|
||||
song_items = self.songs.items()
|
||||
|
||||
lowest_score = min(song_items, key=lambda item: item[1].user_score)[1].user_score
|
||||
highest_score = min(song_items, key=lambda item: item[1].user_score)[1].user_score
|
||||
|
||||
rank += translate(song.user_score, lowest_score, highest_score, 0.0, USER_SCORE_WEIGHT)
|
||||
|
||||
# last10items = self.history[-10:]
|
||||
|
||||
return rank
|
||||
|
||||
|
||||
def translate(value: float, in_min: float, in_max: float, out_min: float, out_max: float):
|
||||
return (value - in_min) / (in_max - in_min) * (out_max - out_min) + out_min
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue