module2 / metrics_v2.json
soyuncj's picture
Upload metrics_v2.json with huggingface_hub
d3b3523 verified
Raw
History Blame Contribute Delete
1.76 kB
{
"columns": [
"f_effective_cosine",
"f_profile_cosine",
"f_profile_l2",
"f_authored_cosine",
"f_authored_available",
"f_liked_cosine",
"f_liked_available",
"f_liked_authored_cosine",
"f_tag_overlap",
"f_tag_jaccard",
"f_common_friend_count",
"f_age_diff",
"f_age_available",
"f_age_band_match",
"f_ui_mode_match"
],
"params": {
"objective": "lambdarank",
"metric": "ndcg",
"n_estimators": 1000,
"learning_rate": 0.05,
"num_leaves": 63,
"min_data_in_leaf": 20,
"random_state": 42,
"ndcg_eval_at": [
5,
10
]
},
"seed": 42,
"n_users": 10000,
"train_pairs": 68193,
"test_pairs": 16741,
"metrics": {
"v2_full": {
"ndcg@5": 0.7721639322500758,
"ndcg@10": 0.8196806119527938,
"n_features": 15
},
"v2_no_content": {
"ndcg@5": 0.7646390549608202,
"ndcg@10": 0.8115972042160688,
"n_features": 10
},
"legacy_embedding": {
"ndcg@5": 0.39210847212656913,
"ndcg@10": 0.4863242966922078,
"n_features": 2
}
},
"gain_importance": {
"f_age_diff": 48362.62074907869,
"f_effective_cosine": 29555.275228187442,
"f_common_friend_count": 29013.391211405396,
"f_profile_cosine": 11398.917454659939,
"f_profile_l2": 10902.10582562536,
"f_authored_cosine": 9033.069555059075,
"f_liked_authored_cosine": 8412.828127481043,
"f_liked_cosine": 7789.788809880614,
"f_age_band_match": 5642.721453994513,
"f_tag_overlap": 4623.446751818061,
"f_age_available": 1804.2311368137598,
"f_ui_mode_match": 960.5918862745166,
"f_authored_available": 65.97200594842434,
"f_liked_available": 42.42172517627478,
"f_tag_jaccard": 0.0
}
}