Learning Assembly Features for Robotic Assembly