mawa00006 / Doping-Detection-Based-on-Publicly-Available-Competition-Data-in-Professional-Road-Cycling

0 stars 0 forks source link

LSTM code #30

Open mawa00006 opened 2 years ago

mawa00006 commented 2 years ago

Screenshot 2022-06-28 at 20 26 04

@tony-hong @sama25100 @SimonP07 I Implemented a LSTM in Tensorflow and trained it using season data of 50 doped and 50 clean riders with a 75/25 train test split. This is the result of 150 training epochs. I think it looks very promising!

In Tensorflow we can easily bias the loss function to account for the data imbalance. I already used this functionality here.

I will try to deploy the model on the computing cluster later in the evening