Thank you @Tejan_Mehndiratta
According to the documentation 2, “input_size” should be an integer. As I stated above, if I set “n_features” = 2, it works without a problem. However, I think that I should be able to set it to 1, since my training data has only one feature column. Setting it 1 causes the error.