Diff of /train_test_split.py [000000] .. [7a2365]

Switch to unified view

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