Hyunji

Delete split_to_test_validation.py

1 -# -*- coding: utf-8 -*-
2 -"""split_to_test_validation.ipynb
3 -
4 -Automatically generated by Colaboratory.
5 -
6 -Original file is located at
7 - https://colab.research.google.com/drive/16Rn0IDJTE6vJMOZtafDAI19AvGD9eK4C
8 -"""
9 -
10 -import pandas as pd
11 -from sklearn.model_selection import train_test_split
12 -
13 -dataset_url = '/content/drive/MyDrive/2d-slice-set-networks-for-brain-age-master/data/test0609.csv'
14 -data = pd.read_csv(dataset_url)
15 -print(data.head())
16 -
17 -y = data.age_at_scan
18 -X = data
19 -
20 -X_train, X_test, y_train, y_test = train_test_split(X, y,test_size=0.3333)
21 -print("\nX_train:\n")
22 -print(X_train.head())
23 -print(X_train.shape)
24 -X_train.to_csv('/content/drive/MyDrive/2d-slice-set-networks-for-brain-age-master/data/test0609.csv', index=False)
25 -
26 -print("\nX_test:\n")
27 -print(X_test.head())
28 -print(X_test.shape)
29 -X_test.to_csv('/content/drive/MyDrive/2d-slice-set-networks-for-brain-age-master/data/valid0609.csv', index=False)
...\ No newline at end of file ...\ No newline at end of file