mantel/src/models/common.py
2024-06-11 21:08:08 +10:00

8 lines
209 B
Python

import numpy as np
from numpy.typing import NDArray
import tempfile
import mlflow
import os
def initialise_random_tensor(*dims: int) -> NDArray[np.float32]:
return np.random.rand(*dims).astype(np.float32)