Upload domainnet/train_validation_split.py with huggingface_hub

domainnet/train_validation_split.py

@@ -0,0 +1,15 @@
+import random
+domains = ["clipart", "infograph", "painting", "quickdraw", "real", "sketch"]
+
+random.seed(42)
+for domain in domains:
+    with open(domain + '_train.txt', 'r') as f:
+        domain_data = f.readlines()
+        random.shuffle(domain_data)
+        valid_data = domain_data[0:int(len(domain_data)*0.2)]
+        train_data = domain_data[int(len(domain_data)*0.2):]
+
+        with open(domain + '_train_fold.txt', 'w') as f:
+            f.write("".join(train_data))
+        with open(domain + '_valid_fold.txt', 'w') as f:
+            f.write("".join(valid_data))