Skip to content

Latest commit

 

History

14 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 

Repository files navigation

🧠 AI-Powered Mental Health Analysis in Tech

GitHub stars GitHub forks GitHub issues GitHub language GitHub last commit

Analyzing workplace psychosocial factors and predicting mental health in the tech industry using machine learning.

📖 Overview

This project undertakes a comprehensive data science investigation into mental health within the tech workplace. Leveraging the OSMI Mental Health in Tech Survey dataset, it aims to analyze various psychosocial factors and build predictive models to identify individuals at risk of mental health conditions. Through rigorous data preprocessing, feature engineering, and a comparative study of machine learning algorithms, the project provides insights into the intricate relationship between work environment and mental well-being, offering a data-driven approach to understanding and potentially mitigating mental health challenges in the technology sector.

✨ Key Analytical Steps & Features

  • Data Ingestion & Preprocessing: Efficiently loading the survey.csv dataset and performing cleaning, handling missing values, and encoding categorical features.
  • Exploratory Data Analysis (EDA): In-depth statistical analysis and visualization of workplace psychosocial factors to uncover patterns, correlations, and key insights.
  • Feature Engineering: Creating new, more informative features from raw data to improve model performance and interpretability.
  • Model Selection & Training: Implementation and training of multiple classification models, including Logistic Regression, Random Forest, and Support Vector Machines (SVM).
  • Hyperparameter Tuning: Optimization of model parameters using GridSearchCV to achieve the best possible predictive performance.
  • Model Evaluation & Comparison: Comprehensive evaluation of trained models using appropriate metrics and comparison of their performance to determine the most effective approach for mental health prediction.

🛠️ Tech Stack

Core Technologies: Python Jupyter Notebook

Data Manipulation & Analysis: Pandas NumPy

Machine Learning: Scikit-learn

Visualization: Matplotlib Seaborn

🚀 Quick Start

To set up the environment and run the analysis notebook, follow these steps:

Prerequisites

  • Python 3.x: Ensure you have Python 3.9 or newer installed.
  • Jupyter Notebook: The primary environment for running the analysis.

Installation

  1. Clone the repository

    git clone https://github.com/zeynpakn/mental-health-tech-analysis.git
    cd mental-health-tech-analysis
  2. Install dependencies It's recommended to create a virtual environment first:

    python -m venv venv
    source venv/bin/activate  # On Windows use `venv\Scripts\activate`

    Then install the required Python packages:

    pip install pandas numpy scikit-learn matplotlib seaborn jupyter

Running the Analysis

  1. Start Jupyter Notebook From the project root directory, launch Jupyter:

    jupyter notebook
  2. Open the notebook Your browser will open to the Jupyter interface. Click on VA_project.ipynb to open the analysis notebook.

  3. Execute cells Run the cells sequentially within the notebook to perform data preprocessing, analysis, model training, and evaluation.

📁 Project Structure

mental-health-tech-analysis/
├── VA_project.ipynb     # Main Jupyter Notebook for data analysis and ML
└── survey.csv           # Dataset: OSMI Mental Health in Tech Survey
└── README.md           

⚙️ Configuration

All configuration, including data loading, feature engineering parameters, and model hyperparameters, is managed directly within the VA_project.ipynb notebook. Users can modify the notebook cells to adjust parameters, experiment with different approaches, or extend the analysis.

🤝 Contributing

While this project is primarily an analytical study, contributions in the form of improved analysis techniques, additional visualizations, or alternative machine learning models are welcome. Please fork the repository and submit a pull request with your suggested changes.

🙏 Acknowledgments

  • OSMI Mental Health in Tech Survey: The invaluable dataset that forms the foundation of this analysis.
  • Python Community: For the robust ecosystem of libraries (Pandas, NumPy, Scikit-learn, Matplotlib, Seaborn) that make such analyses possible.

📞 Support & Contact

  • 🐛 Issues: Feel free to report any issues or suggest improvements via GitHub Issues.

⭐ Star this repo if you find this analysis insightful!

Made with ❤️ by zeynpakn & ozdogrumerve & Hkubrau

powered by caffeine, confusion, and cross-validation

About

AI-powered mental health prediction project analyzing workplace psychosocial factors using machine learning. Includes data preprocessing, feature engineering, statistical analysis, and model comparison with Logistic Regression, Random Forest, SVM, and GridSearchCV on the OSMI Mental Health in Tech Survey dataset.

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages