relationship-longevity-predictor / ensemble_config.json
Builder-Neekhil's picture
Upload ensemble_config.json with huggingface_hub
27f04e7 verified
{
"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
}
}