Skip to content

🧠 Machine learning analysis for the paper named "Predicting 3-year persistent or recurrent major depressive episode using machine learning techniques".

Notifications You must be signed in to change notification settings

brunomontezano/mdd-prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Analysis scripts from "Predicting 3-year persistent or recurrent major depressive episode using machine learning techniques"

DOI DOI

This repository maintains the data analysis from the paper entitled "Predicting 3-year persistent or recurrent major depressive episode using machine learning techniques".

The study aimed to predict major depressive disorder (MDD) recurrence and severe depression in MDD patients after three years of follow-up.

The paper is published at Psychiatry Research Communications.

You can click the badge at the top of README to read the article, or click here.

What does the repo contain?

It contains:

  • Scripts for each prediction model:
    • Recurrent or persistent depression;
    • Severity level of depression among all subjects;
    • Severe depression among depressed subjects.
  • Scripts that generated paper figures;
  • Script that generated paper descriptive table.

The scripts use brazilian portuguese on many sections and variable names. Some files have comments in english to ease the reading for english speakers.

Python libraries used:

The following Python libraries were used: matplotlib, seaborn, pandas.

R packages used:

The following R packages were used: dplyr, caret, pROC, purrr, glmnet, table1, and ggplot2.

Contact

You can contact me at [email protected]