I went for the restricted licence too - for the same reason as you, it was a lot cheaper doing it on my own 125. Although I'd never pass it now on my CBR, I still can't u-turn that thing ;D

You can restrict pretty much any bike, so rather than thinking about engine size I'd recommend sitting on lots of bikes and seeing which one you feel most comfortable on. Then you can work out how to get it restricted.