This commit is contained in:
Ali Al-Dhamen 2023-05-22 17:39:30 -05:00 committed by GitHub
commit 4b86bd2a12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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. # TODO: Read these from command line arguments, since they specify the existing example weights in the input data.
DEFAULT_WEIGHT_BY_LABEL = { DEFAULT_WEIGHT_BY_LABEL = {
"is_clicked": 0.3, "is_clicked": 0.3,
"is_favorited": 1.0, "is_favorited": 1.1,
"is_open_linked": 0.1, "is_open_linked": 0.2,
"is_photo_expanded": 0.03, "is_photo_expanded": 0.1,
"is_profile_clicked": 1.0, "is_profile_clicked": 1.0,
"is_replied": 9.0, "is_replied": 10.0,
"is_retweeted": 1.0, "is_retweeted": 1.5,
"is_video_playback_50": 0.01 "is_video_playback_50": 0.05,
"is_tweet_bookmarked": 1
} }
def add_weight_arguments(parser): def add_weight_arguments(parser):