22 lines
614 B
Python
22 lines
614 B
Python
# checkstyle: noqa
|
|
|
|
INDEX_BY_LABEL = {
|
|
"is_clicked": 1,
|
|
"is_favorited": 2,
|
|
"is_open_linked": 3,
|
|
"is_photo_expanded": 4,
|
|
"is_profile_clicked": 5,
|
|
"is_replied": 6,
|
|
"is_retweeted": 7,
|
|
"is_video_playback_50": 8
|
|
}
|
|
|
|
TARGET_LABEL_IDX = 0
|
|
EB_SCORE_IDX = 9
|
|
|
|
LABEL_NAMES = [label_name for label_name, _ in sorted(INDEX_BY_LABEL.items(), key=lambda item: item[1])]
|
|
|
|
PREDICTED_CLASSES = \
|
|
["tf_target"] + ["tf_" + label_name for label_name in LABEL_NAMES] + ["tf_timelines.earlybird_score"] + \
|
|
["lolly_target"] + ["lolly_" + label_name for label_name in LABEL_NAMES] + ["lolly_timelines.earlybird_score"]
|