text_plain_output_1.png image_output_1.png
from radtorch import pipeline, core, utils

train_dir = '/train_data/train/'
test_dir = '/test_data/test1/'

table = utils.datatable_from_filepath(train_dir, classes=['dog', 'cat'])

clf = pipeline.Image_Classification(data_directory=train_dir, is_dicom=False, table=table, type='nn_classifier', model_arch='vgg16', epochs=10, batch_size=100, sampling=0.15)

clf.data_processor.dataset_info(plot=False)