Great, this seems like a better place to initialize `b` than where it was previously (in `__init__`), because even if `b` is generally not affected by the input space, it was confusing that `set_input_space` was resetting `W` but not `b`.