πŸ“± Social Media Usage Risk Predictor

A Machine Learning web application that predicts Social Media Usage Risk Level (Low / Medium / High) based on user behavioral and demographic inputs.

🎯 Overview

This app uses a Random Forest Classifier trained on an E-commerce Customer Behavior dataset to classify users into three risk categories based on features like:

  • Age & Gender
  • City & Membership Type
  • Total Spend & Items Purchased
  • Average Rating & Satisfaction Level
  • Days Since Last Purchase
  • Discount Applied

πŸ€– Model Details

Property Value
Algorithm Random Forest Classifier
Estimators 150
Max Depth 8
Training Samples 1,200
Features 10
Classes Low / Medium / High

πŸš€ How to Run Locally

git clone https://huggingface.co/spaces/<your-username>/social-media-risk-predictor
cd social-media-risk-predictor
pip install -r requirements.txt
streamlit run app.py

πŸ“Š Dataset

Based on the E-commerce Customer Behavior Dataset.

Features include: Customer ID, Gender, Age, City, Membership Type, Total Spend, Items Purchased, Average Rating, Discount Applied, Days Since Last Purchase, and Satisfaction Level.

πŸ›  Tech Stack

  • Python 3.10+
  • Streamlit β€” Web interface
  • Scikit-learn β€” ML model (Random Forest)
  • Pandas / NumPy β€” Data processing
  • Hugging Face Spaces β€” Deployment

πŸ“Έ Screenshot

App Screenshot


Built with ❀️ for ML deployment demonstration.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support