File size: 4,325 Bytes
27f04e7 | 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 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | {
"weights": {
"xgboost": 0.4,
"lightgbm": 0.35,
"catboost": 0.25
},
"optimal_threshold": 0.16446829069780428,
"feature_columns": [
"dater_age",
"dated_age",
"age_difference",
"same_race_importance_for_dater",
"same_religion_importance_for_dater",
"attractiveness_importance_for_dated",
"sincerity_importance_for_dated",
"intelligence_importance_for_dated",
"humor_importance_for_dated",
"ambition_importance_for_dated",
"shared_interests_importance_for_dated",
"attractiveness_score_of_dater_from_dated",
"sincerity_score_of_dater_from_dated",
"intelligence_score_of_dater_from_dated",
"humor_score_of_dater_from_dated",
"ambition_score_of_dater_from_dated",
"shared_interests_score_of_dater_from_dated",
"attractiveness_importance_for_dater",
"sincerity_importance_for_dater",
"intelligence_importance_for_dater",
"humor_importance_for_dater",
"ambition_importance_for_dater",
"shared_interests_importance_for_dater",
"self_reported_attractiveness_of_dater",
"self_reported_sincerity_of_dater",
"self_reported_intelligence_of_dater",
"self_reported_humor_of_dater",
"self_reported_ambition_of_dater",
"reported_attractiveness_of_dated_from_dater",
"reported_sincerity_of_dated_from_dater",
"reported_intelligence_of_dated_from_dater",
"reported_humor_of_dated_from_dater",
"reported_ambition_of_dated_from_dater",
"reported_shared_interests_of_dated_from_dater",
"dater_interest_in_sports",
"dater_interest_in_tvsports",
"dater_interest_in_exercise",
"dater_interest_in_dining",
"dater_interest_in_museums",
"dater_interest_in_art",
"dater_interest_in_hiking",
"dater_interest_in_gaming",
"dater_interest_in_clubbing",
"dater_interest_in_reading",
"dater_interest_in_tv",
"dater_interest_in_theater",
"dater_interest_in_movies",
"dater_interest_in_concerts",
"dater_interest_in_music",
"dater_interest_in_shopping",
"dater_interest_in_yoga",
"interests_correlation",
"expected_satisfaction_of_dater",
"expected_number_of_likes_of_dater_from_20_people",
"expected_number_of_dates_for_dater",
"dater_liked_dated",
"probability_dated_wants_to_date",
"attractiveness_perception_gap",
"attractiveness_mutual_score",
"attractiveness_perception_product",
"sincerity_perception_gap",
"sincerity_mutual_score",
"sincerity_perception_product",
"intelligence_perception_gap",
"intelligence_mutual_score",
"intelligence_perception_product",
"humor_perception_gap",
"humor_mutual_score",
"humor_perception_product",
"ambition_perception_gap",
"ambition_mutual_score",
"ambition_perception_product",
"attractiveness_value_fulfillment_dater",
"sincerity_value_fulfillment_dater",
"intelligence_value_fulfillment_dater",
"humor_value_fulfillment_dater",
"ambition_value_fulfillment_dater",
"attractiveness_self_awareness_gap",
"sincerity_self_awareness_gap",
"intelligence_self_awareness_gap",
"humor_self_awareness_gap",
"ambition_self_awareness_gap",
"total_perception_gap",
"total_mutual_score",
"total_value_fulfillment",
"total_self_awareness_gap",
"expectation_meets_reality",
"confidence_calibration",
"age_gap_abs",
"age_gap_squared",
"dater_is_older",
"combined_age",
"interest_diversity",
"interest_intensity",
"max_interest",
"min_interest",
"interest_range",
"importance_concentration_dater",
"max_importance_dater",
"importance_concentration_dated",
"importance_alignment_0",
"importance_alignment_1",
"importance_alignment_2",
"importance_alignment_3",
"importance_alignment_4",
"importance_alignment_5",
"total_importance_alignment",
"dater_race_encoded",
"dated_race_encoded",
"race_match",
"is_dater_male_int",
"are_same_race_int",
"already_met_int"
],
"model_files": {
"xgboost": "xgboost_model.joblib",
"lightgbm": "lightgbm_model.joblib",
"catboost": "catboost_model.cbm"
},
"metrics": {
"auc_roc": 0.8911758101938478,
"auc_pr": 0.6984966289637278,
"f1": 0.6350710900473934,
"accuracy": 0.8530534351145038
}
} |