in train.py [0:0]
def setup_model_artifacts(artifactpath, model_name):
def try_mkdir(dirname):
try:
os.mkdir(dirname)
except:
pass
try_mkdir(f"{artifactpath}/")
try_mkdir(f"{artifactpath}/{model_name}")