Jérôme BUISINE
|
c361cbf571
fix params type issue
|
4 gadi atpakaļ |
Jérôme BUISINE
|
e1296990b8
use of params to fit well images input of model
|
4 gadi atpakaļ |
Jérôme BUISINE
|
5ee7f333f1
roc auc score test change input values
|
4 gadi atpakaļ |
Jérôme BUISINE
|
1e89cf4438
fix evaluate
|
4 gadi atpakaļ |
Jérôme BUISINE
|
82da620360
fix issue when using metric
|
4 gadi atpakaļ |
Jérôme BUISINE
|
74a168da04
predict test input update
|
4 gadi atpakaļ |
Jérôme BUISINE
|
4ebed7df49
add output info
|
4 gadi atpakaļ |
Jérôme BUISINE
|
91339204bd
list of prediction
|
4 gadi atpakaļ |
Jérôme BUISINE
|
59473be88f
try convert to tensor
|
4 gadi atpakaļ |
Jérôme BUISINE
|
25b939efe3
change predict input type
|
4 gadi atpakaļ |
Jérôme BUISINE
|
70f5d74832
- add models backups for LSTM
|
4 gadi atpakaļ |
Jérôme BUISINE
|
567da49b12
add class weight for cnn model
|
4 gadi atpakaļ |
Jérôme BUISINE
|
ae38dc94c9
Split whole data into train, val and test
|
4 gadi atpakaļ |
Jérôme BUISINE
|
a7545babe8
Prediction scripts added for CNN and LSTM
|
4 gadi atpakaļ |
Jérôme BUISINE
|
e9609c0d47
add first version of LSTM 2D model
|
4 gadi atpakaļ |
Jérôme BUISINE
|
a8ebfc08e9
Dataset generation updates
|
4 gadi atpakaļ |
Jérôme BUISINE
|
8bcaf1881f
Add of RGB image as input for 3D model
|
5 gadi atpakaļ |
Jérôme BUISINE
|
288e5df499
Update of train model process and modules
|
5 gadi atpakaļ |
Jérôme BUISINE
|
f923d55c3d
Update of modules and training data
|
5 gadi atpakaļ |
Jérôme BUISINE
|
2567770dc9
Update of auc train score into backup model
|
5 gadi atpakaļ |
Jérôme BUISINE
|
50019849be
Update of prediction script
|
5 gadi atpakaļ |
Jérôme BUISINE
|
1339e90bb1
Update of resuming computation
|
5 gadi atpakaļ |
Jérôme BUISINE
|
3369b67e76
Add of validation split info
|
5 gadi atpakaļ |
Jérôme BUISINE
|
a6d566fc00
Update of way of generating data and train model
|
5 gadi atpakaļ |
Jérôme BUISINE
|
90f747bedf
Update of validation split into train model
|
5 gadi atpakaļ |
Jérôme BUISINE
|
add1a54332
Update of whole project for calculco
|
5 gadi atpakaļ |
Jérôme BUISINE
|
82663444c4
Update of checkpoint model
|
5 gadi atpakaļ |
Jérôme BUISINE
|
1ce9187713
Add of model backup every epochs
|
5 gadi atpakaļ |
Jérôme BUISINE
|
8fc7c020c1
Add of all models into this project and no longer in common modules
|
5 gadi atpakaļ |
Jérôme BUISINE
|
483d4ab0af
Test of transfer learning network
|
5 gadi atpakaļ |