Check out this site. It is a program that gives you soldiers to train to do things using advanced AI, and put milestones after they have learnt someting to make them remember.
For example, shoot them...then punish them for being shot. they will learn to not get shot. Then you put a milestone in .