Add files via upload

jim-schwoebel · Aug 14, 2020 · 1c3ebbc · 1c3ebbc
1 parent f13388f
commit 1c3ebbc
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/train_dir/create_dataset.py b/train_dir/create_dataset.py
@@ -0,0 +1,13 @@
+import pandas as pd
+import os, shutil
+
+data=pd.read_csv('pga1_lupus.csv')
+urls=(data['url'])
+sleqols=list(data['pga1_lupus_active'])
+os.mkdir('pga1_lupus_active_above5.65')
+os.mkdir('pga1_lupus_active_below5.65')
+for i in range(len(data)):
+	if sleqols[i] > 5.65:
+		shutil.copy(urls[i], os.getcwd()+'/'+'pga1_lupus_active_above5.65'+'/'+urls[i].split('/')[-1])
+	else:
+		shutil.copy(urls[i], os.getcwd()+'/'+'pga1_lupus_active_below5.65'+'/'+urls[i].split('/')[-1])