-
Notifications
You must be signed in to change notification settings - Fork 0
/
cleanup.py
48 lines (32 loc) · 936 Bytes
/
cleanup.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
44
45
46
47
48
import pymongo
import shutil
import json
import os
from spearmint.utils.parsing import parse_db_address
def main():
dev_path = os.environ['DNN_PATH']
try:
shutil.rmtree( '%s/output' % dev_path )
print 'deleted: %s/output' % dev_path
except:
pass
try:
shutil.rmtree( '%s/log' % dev_path )
print 'deleted: %s/log' % dev_path
except:
pass
drop_db(dev_path)
def drop_db(path):
if not os.path.isdir(path):
raise Exception("%s is not a valid directory" % path)
with open(os.path.join(path, 'config.json'), 'r') as f:
cfg = json.load(f)
db_address = parse_db_address(cfg)
print 'Cleaning up experiment %s in database at %s' % (cfg["experiment-name"], db_address)
client = pymongo.MongoClient(db_address)
db = client.spearmint
db[cfg["experiment-name"]]['jobs'].drop()
db[cfg["experiment-name"]]['hypers'].drop()
del client
if __name__ == '__main__':
main()