YOONHO-CHOI / BTcP-prediction

Home Page:https://github.com/YOONHO-CHOI/BTcP-prediction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Break through cancer pain (BTcP) prediction

Breakthrough cancer pain (BTcP) is a challenging clinical problem in managing cancer pain.

We investigated the clinical relevance of deep neural network models that predicts the onset time of BTcP.

BTcP was defined as the pain with a numerical rating scale (NRS) score of 4 or above.

Model

The model consisted of three basic LSTM blocks stacked followed by dense layers.

The model was trained for 300 epochs with a batch size of 100 with balanced cross entropy loss and was optimized by stochastic weight averaging (SWA) with an initial learning rate of 1e-4, start averaging of 5, and the average period of 1. Our model was programmed in Python 3.7, Tensorflow 2.4.1 version, and experimented with NVIDIA Geforce RTX 2080.

Evaluation metric

Matthews correlation coefficient (MCC)

Results

Cases of prediction for onset timing of BTcP

About

https://github.com/YOONHO-CHOI/BTcP-prediction

License:MIT License


Languages

Language:Python 100.0%