-
Notifications
You must be signed in to change notification settings - Fork 0
/
addData.py
43 lines (42 loc) · 1.39 KB
/
addData.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import config
import csv
import json
import os
import subprocess
import sys
import train
args = sys.argv
if len(args) < 2:
print("Two arguments required")
elif len(args) > 2:
print("Too many arguments")
else:
# Open status file and load the current batch size
with open(config.STATUSPATH, 'r') as statusFile:
status = json.load(statusFile)
currSize = status["currSize"]
numOfData = int(args[1])
if numOfData == 1 or numOfData == 2:
# Adding data
with open(config.DATAPATH, 'a') as csvFile:
writer = csv.writer(csvFile)
writer.writerows(config.FAKEDATA[:numOfData])
# Decide if training is required
if currSize + numOfData >= config.BATCHSIZE:
status["currSize"] = 0
modelPath = train.train()
if status["host"] == "blue":
subprocess.run(["sh", "deploy.sh", "green", modelPath])
status["host"] = "green"
elif status["host"] == "green":
subprocess.run(["sh", "deploy.sh", "blue", modelPath])
status["host"] = "blue"
else:
print("No host name: " + status["host"])
else:
status["currSize"] += numOfData
# Update status file
with open(config.STATUSPATH, 'w') as statusFile:
json.dump(status, statusFile)
else:
print("Not supported yet")