update train notebook #201

Closed felenitaribeiro closed 1 year ago