|
a |
|
b/train_test_split.py |
|
|
1 |
# -*- coding: utf-8 -*- |
|
|
2 |
"""Train_test_split.ipynb |
|
|
3 |
|
|
|
4 |
** |
|
|
5 |
* This file is part of Hybrid CNN-LSTM for COVID-19 Severity Score Prediction paper. |
|
|
6 |
* |
|
|
7 |
* Written by Ankan Ghosh Dastider and Farhan Sadik. |
|
|
8 |
* |
|
|
9 |
* Copyright (c) by the authors under Apache-2.0 License. Some rights reserved, see LICENSE. |
|
|
10 |
*/ |
|
|
11 |
|
|
|
12 |
""" |
|
|
13 |
|
|
|
14 |
get_ipython().__class__.__name__ = "ZMQInteractiveShell" |
|
|
15 |
|
|
|
16 |
print(X_Train.shape) |
|
|
17 |
print(Y_train.shape) |
|
|
18 |
|
|
|
19 |
BATCH_SIZE =64 |
|
|
20 |
|
|
|
21 |
|
|
|
22 |
# Split the train and validation sets |
|
|
23 |
#X_train, X_val, Y_train, Y_val = train_test_split(Final_X, Y_train, test_size=0.2, random_state=SEED) |
|
|
24 |
X_train, X_val, Y_train, Y_val = train_test_split(X_Train, Y_train, test_size=0.2, random_state=SEED) |
|
|
25 |
|