File size: 2,373 Bytes
7daf628 | 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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | """Loggers."""
import os
from os.path import dirname, realpath, abspath
from tqdm.auto import tqdm
import numpy as np
import json
import yaml
curr_filepath = abspath(__file__)
repo_path = dirname(dirname(dirname(curr_filepath)))
# repo_path = dirname(dirname(dirname(realpath(__file__))))
def tqdm_iterator(items, desc=None, bar_format=None, **kwargs):
tqdm._instances.clear()
iterator = tqdm(
items,
desc=desc,
# bar_format='{l_bar}{bar:10}{r_bar}{bar:-10b}',
**kwargs,
)
tqdm._instances.clear()
return iterator
def set_tqdm_pandas(desc=None, total=None):
from tqdm import tqdm
tqdm.pandas(
desc=desc, total=total,
bar_format="{l_bar}{bar:30}{r_bar}{bar:-30b}",
)
def print_retrieval_metrics_for_csv(metrics, scale=100):
print_string = [
np.round(scale * metrics["R1"], 3),
np.round(scale * metrics["R5"], 3),
np.round(scale * metrics["R10"], 3),
]
if "MR" in metrics:
print_string += [metrics["MR"]]
print()
print("Final metrics: ", ",".join([str(x) for x in print_string]))
print()
def get_terminal_width():
import shutil
return shutil.get_terminal_size().columns
def print_update(update, fillchar=".", color="yellow", pos="left", **kwargs):
from termcolor import colored
# add ::: to the beginning and end of the update s.t. the total length of the
# update spans the whole terminal
try:
terminal_width = get_terminal_width()
except:
terminal_width = 98
if pos == "center":
update = update.center(len(update) + 2, " ")
update = update.center(terminal_width, fillchar)
elif pos == "left":
update = update.ljust(terminal_width, fillchar)
update = update.ljust(len(update) + 2, " ")
elif pos == "right":
update = update.rjust(terminal_width, fillchar)
update = update.rjust(len(update) + 2, " ")
else:
raise ValueError("pos must be one of 'center', 'left', 'right'")
print(colored(update, color, **kwargs))
def json_print(data, indent=4):
print(json.dumps(data, indent=indent))
def disp_dict(data):
print("-" * get_terminal_width())
print(yaml.dump(data, default_flow_style=False))
print("-" * get_terminal_width())
if __name__ == "__main__":
print("Repo path:", repo_path) |