Skip to content

Xueping/BiteNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BiteNet

Data Preparation

dataset.data_prepararion.py for MIMIC III dataset

Train Model

BiteNet.train.BiteNet_mh_DX.py for future diagnosis prediction.

BiteNet.train.BiteNet_mh_RE.py for future re-admission prediction.

Model Parameters

--data_source mimic3

--model Bite

--verbose True

--task BiteNet

--predict_type dx

--visit_threshold 2

--max_epoch 5

--train_batch_size 32

--gpu 2

--valid_visits 10

--num_hidden_layers 1

--pos_encoding encoding

--min_cut_freq 5

--embedding_size 128

--dropout 0.1

--only_dx_flag False

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages