def save_model( model,fileDescription, modelName=''):
if not os.path.exists(getPath('model')):
os.makedirs(getPath('model'))
modelName = makeModelName(fileDescription,modelName)
print('save model as %s' % modelName)
modelName = '%s/%s'%(getPath('model'),modelName)
joblib.dump(model, modelName)
return modelName
def load_model(fileDescription,modelName=''):
modelName = makeModelName(fileDescription, modelName)
modelName = '%s/%s' % (getPath('model'), modelName)
print('load --> %s'%modelName)
model = joblib.load(modelName)
return model
def saveFig(axes,figName):
if len(figName)>0:
axes.savefig('%s/%s.png'%(getPath('fig'),figName))
