Classtree is a hierarchical classifier for images or text.
pip install classtree
The fastest way to use Classtree is to call the CLI on a folder of images or text files.
train_data/
|- animals/
|- mammals/
|- marsupials/
|- koala/
|- image001.jpg
|- image002.jpg
|- ...
|- ...
|- reptiles/
|- ...
|- ...
|- ...
classtree train images --model animals --dir train_data/animals
or
classtree train text --model animals --dir train_data/animals
And then use your model with the predict command.
classtree predict --model animals new_data/image304.jpg
> birds/raptors/eagle
You can download a pre-trained model using the download command.
classtree download --model dbpedia
Or download a pre-prepared dataset.
classtree download --text dbpedia
If you want to fine-tune an existing model, you can use the --from
flag during training with any downloaded model.
classtree train text --model animals --from dbpedia --dir train_data/animals
Task | Name | Size | Dataset | Notes |
---|---|---|---|---|
Text Classification | dbpedia | M | dbpedia |
Type | Name | Dataset | Notes |
---|---|---|---|
Image | inaturalist21-mini | iNaturalist 2021 (Mini) | Non-commercial research/educational use only |
Text | dbpedia | DBPedia | CC0: Public Domain |
You can test your model on a hold-out dataset using the test
command.
classtree test --model animals --dir=test_data/animals
Classtree is available for non-commercial internal research use by academic institutions or not-for-profit organisations only, free of charge. Please, see the license for further details. To the extent permitted by applicable law, your use is at your own risk and our liability is limited. Interested in a commercial license? For commercial queries, please email aimlshop@adelaide.edu.au with subject line “Classtree Commercial License”.
This is an AIML Shop project.