35 lines
881 B
Python
35 lines
881 B
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
'''
|
|
Author: John
|
|
Email: johnjim0816@gmail.com
|
|
Date: 2021-03-12 16:02:24
|
|
LastEditor: John
|
|
LastEditTime: 2021-04-29 15:32:38
|
|
Discription:
|
|
Environment:
|
|
'''
|
|
import os
|
|
import numpy as np
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
def save_results(rewards,ma_rewards,tag='train',path='./results'):
|
|
'''保存reward等结果
|
|
'''
|
|
np.save(path+'{}_rewards.npy'.format(tag), rewards)
|
|
np.save(path+'{}_ma_rewards.npy'.format(tag), ma_rewards)
|
|
print('results saved!')
|
|
|
|
def make_dir(*paths):
|
|
for path in paths:
|
|
Path(path).mkdir(parents=True, exist_ok=True)
|
|
def del_empty_dir(*paths):
|
|
'''del_empty_dir delete empty folders unders "paths"
|
|
'''
|
|
for path in paths:
|
|
dirs = os.listdir(path)
|
|
for dir in dirs:
|
|
if not os.listdir(os.path.join(path, dir)):
|
|
os.removedirs(os.path.join(path, dir)) |