import numpy as np def prepare_model(n, p): """ p must be between 0 and 1. """ return (np.random.rand(n, n) < p).astype(np.int8)