This commit is contained in:
Ali Al-Dhamen 2023-07-17 21:40:38 -05:00 committed by GitHub
commit 66382eec2d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 6 deletions

View File

@ -5,13 +5,14 @@ from .constants import INDEX_BY_LABEL, LABEL_NAMES
# TODO: Read these from command line arguments, since they specify the existing example weights in the input data.
DEFAULT_WEIGHT_BY_LABEL = {
"is_clicked": 0.3,
"is_favorited": 1.0,
"is_open_linked": 0.1,
"is_photo_expanded": 0.03,
"is_favorited": 1.1,
"is_open_linked": 0.2,
"is_photo_expanded": 0.1,
"is_profile_clicked": 1.0,
"is_replied": 9.0,
"is_retweeted": 1.0,
"is_video_playback_50": 0.01
"is_replied": 10.0,
"is_retweeted": 1.5,
"is_video_playback_50": 0.05,
"is_tweet_bookmarked": 1
}
def add_weight_arguments(parser):