File size: 1,763 Bytes
d3b3523
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
{
  "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
  }
}