engagement-predictor / config.json
raviearjun's picture
Add feature schema config.json
1fdc633 verified
Raw
History Blame Contribute Delete
2.57 kB
{
"task": "tabular-regression",
"library_name": "skops",
"model_file": "engagement_ridge.skops",
"feature_columns": [
"brand_industry",
"brief_required_hook_type",
"content_upload_day",
"brief_required_tone__fun",
"brief_required_tone__calm",
"brief_required_tone__confident",
"brief_required_tone__energetic",
"brief_required_tone__glam",
"brief_required_tone__fresh",
"brief_required_tone__bold",
"brief_required_tone__educational",
"brief_required_tone__active",
"brief_required_tone__aspirational",
"brief_required_tone__scientific",
"brief_required_tone__authentic",
"brief_target_creator_tier__nano",
"brief_target_creator_tier__micro",
"brief_target_creator_tier__mid",
"brief_target_creator_tier__macro",
"brief_campaign_objective__awareness",
"brief_campaign_objective__trial",
"brief_campaign_objective__conversion",
"brief_campaign_objective__engagement",
"brief_target_audience_gender__female",
"brief_target_audience_gender__male",
"brief_visual_style__studio",
"brief_visual_style__natural",
"brief_visual_style__outdoor",
"brief_visual_style__glam_lighting",
"brief_dress_code__casual",
"brief_dress_code__stylish",
"brief_dress_code__activewear",
"brief_dress_code__colorful",
"brief_dress_code__bold_colors",
"brand_avg_campaign_engagement_score",
"brand_n_product_categories",
"brief_product_placement_sec",
"brief_n_required_keywords",
"brief_n_mandatory_hashtags",
"brief_n_mandatory_mentions",
"brief_n_key_messages",
"brief_target_age_min",
"brief_target_age_max",
"brief_target_age_span",
"brief_duration_range_min",
"brief_duration_range_max",
"content_duration_sec",
"content_upload_hour_sin",
"content_upload_hour_cos",
"content_is_weekend",
"content_n_hashtags",
"content_caption_len_chars",
"content_caption_len_words",
"content_caption_n_emoji",
"content_caption_n_hashtags_inline",
"content_caption_n_mentions_inline",
"content_sound_popularity",
"content_hashtag_compliance",
"content_mention_compliance"
],
"target": "engagement_score",
"training_metrics": {
"alpha": 10.0,
"n_rows": 2000,
"n_groups": 300,
"cv_mae": 0.011419824566734465,
"cv_rmse": 0.01458130687031906,
"cv_r2": 0.7102257580449983,
"baseline_mae": 0.021489796999999998,
"baseline_rmse": 0.027087352583401354
},
"sklearn_estimator": "sklearn.pipeline.Pipeline(ColumnTransformer + Ridge)"
}