validate/utils.py

from termcolor import colored
def rprint(*args):
    combined_text = ' '.join(map(str, args))
    print(colored(combined_text, 'white', 'on_red', attrs=["bold"]))

def yprint(*args):
    combined_text = ' '.join(map(str, args))
    print(colored(combined_text, 'yellow',attrs=["bold"]))

def gprint(*args):
    combined_text = ' '.join(map(str, args))
    print(colored(combined_text, 'green',attrs=["bold"]))

def draw_histogram(predictions, targets, save_path):
    import matplotlib
    import matplotlib.pyplot as plt
    from pathlib import Path
    save_path = Path(save_path)
    save_path.parent.mkdir(parents=True, exist_ok=True)

    matplotlib.use('Agg') 
    plt.figure(figsize=(10, 5))

    plt.subplot(1, 2, 1)
    plt.hist(predictions, bins=20, alpha=0.5, color='blue', label='Predictions')
    plt.title('Predictions')
    plt.xlabel('Value')
    plt.ylabel('Frequency')

    plt.subplot(1, 2, 2)
    plt.hist(targets, bins=20, alpha=0.5, color='red', label='Targets')
    plt.title('Targets')
    plt.xlabel('Value')
    plt.ylabel('Frequency')

    plt.tight_layout()
    # plt.show()
    plt.savefig(save_path)