{"cells":[{"metadata":{},"cell_type":"markdown","source":"**To see the traning process please refer to [Traing the model for classification](https://www.kaggle.com/sd4321/latest-fastai-training-inference-prediction)****"},{"metadata":{"_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","collapsed":true,"_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","trusted":false},"cell_type":"code","source":"import pandas as pd\nimport fastai\nfrom fastai.vision.all import *","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"path = Path('../input/cassava-leaf-disease-classification')\nimage_path = Path('../input/cassava-leaf-disease-classification/train_images')\ntraining_data_file = Path('../input/cassava-leaf-disease-classification/train.csv')\nsample_submission_file = Path('../input/cassava-leaf-disease-classification/sample_submission.csv')","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train_df = pd.read_csv(training_data_file)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"src_size = (PILImage.create('../input/cassava-leaf-disease-classification/train_images/1000015157.jpg')).size\nbs = 16\n\ndef get_x(r):return image_path/r['image_id']\n\ndef get_y(r):return r['label']\n\nitem_tfms = [Resize(src_size)]\n\nbatch_tfms = [RandomResizedCrop(size=224), *aug_transforms(mult=1.0, do_flip=True, max_rotate=40.0),Normalize.from_stats(*imagenet_stats)]\n\n\ndatablockk = DataBlock(blocks=(ImageBlock, CategoryBlock),splitter=RandomSplitter(seed=42),get_x=get_x,get_y=get_y,item_tfms = item_tfms,batch_tfms = batch_tfms )\n\nimage_data_loader = datablockk.dataloaders(train_df, bs=bs)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"learn = load_learner('../input/trainedmodel/export (1).pkl')","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"test_image_files = Path('../input/cassava-leaf-disease-classification/test_images')\nimage_data_loader_test = image_data_loader.test_dl(get_image_files(test_image_files))\nlabels = []\nfiles = []\nfor file in list(test_image_files.glob(\"*.jpg\")):\n    files.append(file.name)\n    labels.append(str(learn.predict(file)[0]))\nsub = pd.read_csv(sample_submission_file)\nsub.head()\nsub['label'] = labels\nsub.to_csv('submission.csv', index=False)","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat":4,"nbformat_minor":4}