Go to the train option in your city, click DESIGN to open up unit design window and click NEW. When you design your troops, go to the TRAITS tab and select either chain mail or plate mail proficiency from the blue traits, they are the two bottom ones.