# cellclass > Cell classification using machine learning models ## Docs ### API Reference #### Datasets > Loading labelled crops for training. - [ROIDataset](https://hocheggerlab.github.io/omero-screen/cellclass/reference/ROIDataset.html): Dataset of image regions (ROIs) and integer class labels - [prepare_dataset](https://hocheggerlab.github.io/omero-screen/cellclass/reference/prepare_dataset.html): Prepare an .npz dataset from a directory of .npy cell image files #### Models > Model construction and evaluation. - [Model](https://hocheggerlab.github.io/omero-screen/cellclass/reference/Model.html): Supported classification model architectures - [create_model](https://hocheggerlab.github.io/omero-screen/cellclass/reference/create_model.html): Create a classification model for cell image data - [test_epoch](https://hocheggerlab.github.io/omero-screen/cellclass/reference/test_epoch.html): Run one evaluation pass and return true and predicted labels