Mmh, aren't you supposed to play with regular partitions when you play keys ?
Just a suposition, I'm not specially a musician, but I'd say so.
Anyway, a good thing would be to get Guitar pro 4, there's a graphic neck and a graphic keyboard which shows the notes played.
You can also display the tab and the regular notation at the same time.
So I think it can help a lot.