Train related things to do in Tokyo?

I’m going to Tokyo in April and looking for things to do. I’ve been to the Railway Museum in Saitama before, and I was thinking of going to the Train and Bus Museum. Is there anything else people would recommend?