there is error while training this.
here is the link to my code. I dont understand what is wrong
link- https://drive.google.com/open?id=1Jh-U2Z_Iw25aabSc-_a4WbKmKNF_Dnzq
Chemical segregation challlenge
Hey @shreyanarula3, since you are using vectorization, you need to proceed in a little different way,
Take this as a reference, and try to figure it out. If unable to do so than you may ask doubts.
HOpe this resolved your doubt.
Mark it as resolved as well. 
I hope I’ve cleared your doubt. I ask you to please rate your experience here
Your feedback is very important. It helps us improve our platform and hence provide you
the learning experience you deserve.
On the off chance, you still have some questions or not find the answers satisfactory, you may reopen
the doubt.