Source code for deepr.examples.movielens.macros.paths

"""Macro that generates new path_model based on date."""

import datetime


[docs]class Paths(dict): """Macro that generates new path_model based on date.""" def __init__(self, path_root: str = None, run_name: str = None, **kwargs): now = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S") if path_root is None: path_root = f"viewfs://root/user/deepr/dev/movielens/models/{now}" if run_name is None: run_name = f"movielens.{now}" path_model = f"{path_root}/model" path_embeddings_svd = f"{path_root}/embeddings_svd.npz" path_counts = f"{path_root}/counts.npz" path_init_ckpt = f"{path_root}/init_ckpt" path_mapping = f"{path_root}/data/mapping.txt" path_record_train = f"{path_root}/data/train.tfrecord.gz" path_record_eval = f"{path_root}/data/eval.tfrecord.gz" path_record_test = f"{path_root}/data/test.tfrecord.gz" path_variables = f"{path_root}/variables" path_embeddings = f"{path_variables}/embeddings" path_biases = f"{path_variables}/biases" path_saved_model = f"{path_root}/saved_model" path_predictions = f"{path_root}/predictions.parquet.snappy" super().__init__( run_name=run_name, path_root=path_root, path_model=path_model, path_embeddings_svd=path_embeddings_svd, path_counts=path_counts, path_init_ckpt=path_init_ckpt, path_mapping=path_mapping, path_record_train=path_record_train, path_record_eval=path_record_eval, path_record_test=path_record_test, path_variables=path_variables, path_embeddings=path_embeddings, path_biases=path_biases, path_saved_model=path_saved_model, path_predictions=path_predictions, **kwargs, )