# Generate a prediction
pred = X @ W + B