I’ve also been following @wgpubs whilst attempting the toxic comments challenge and have been enormously helped by their comments. I’m only a little ahead of you in the course and am no expert, but eventually managed to put a multiclass classifier together. Unfortunately I can’t get it to train! So much to learn still…
As I understand it, the only need for a custom classifier is to handle the various softmax / sigmoid layers for the different outputs in the original question. For toxic comments we just need softmax, so I only modified the original PoolingLinearClassifier slightly by adding softmax. Maybe I’m wrong, hopefully somebody will correct me if so.
I’ve put my custom classifier here in case it helps. Custom PLC Gist
Happy to share my whole notebook if you want as there’s a lot of custom code, but since it’s not training I don’t want to send someone else down the wrong path too…