Skip to content

πŸŽ‰ AutoClassification

Pre-release
Pre-release
Compare
Choose a tag to compare
@aniketmaurya aniketmaurya released this 22 Aug 14:07
· 184 commits to main since this release
  • Added AutoImageClassification
  • Added AutoTextClassification

Example

model = AutoImageClassifier(datamodule,
                            suggested_backbones=['ssl_resnet18'],
                            suggested_conf=suggested_conf,
                            max_epochs=1,
                            optimization_metric="val_accuracy",
                            timeout=30)

print("AutoImageClassifier initialised!")
model.hp_tune()