Skip to content

update pytorch classifier

Max Trevor requested to merge mtrevor_update_pytorch into master

PyTorch classifier now balances classes and uses Adam as optimizer. SupervisedSklearnClassifier calculates class weights during training for use by its subclasses

Merge request reports