14 lines
141 B
Python
14 lines
141 B
Python
|
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)
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|