🔥 Prepare ML Environment | Seed All The Things 🔥

set_env.py

import os
import torch
import random
import numpy as np
import pandas as pd

def set_seed(seed: int):
    """
    Set random seed for reproducibility across numpy, torch, and python.
    """
    random.seed(seed)
    np.random.seed(seed)
    torch.manual_seed(seed)

    # if using CUDA
    torch.cuda.manual_seed(seed)
    torch.cuda.manual_seed_all(seed)

    # ensure deterministic behavior
    torch.backends.cudnn.deterministic = True
    torch.backends.cudnn.benchmark = False

    # set PYTHONHASHSEED environment variable
    os.environ['PYTHONHASHSEED'] = str(seed)

    print(f"seeds set to: {seed}")

set_seed(seed=37)

pd.set_option("display.max_columns", None)
pd.set_option('display.max_rows', None)
pd.set_option('display.width', 1000)
pd.set_option("max_colwidth", None)
pd.set_option("max_seq_item", None)