python
from alchemy_annotations import ImageClassifier
classifier = ImageClassifier()
classifier.load_dataset("path/to/dataset")
classifier.train()
result = classifier.predict("path/to/image")
print(result)
yaml
model:
type: "vgg"
pretrained: true
classes: 10
dataset:
train:
dir: "path/to/train"
val:
dir: "path/to/validation"
