{"cells":[{"metadata":{"_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","trusted":true},"cell_type":"code","source":"from fastai.vision.all import *\n\nmodel_path = \"../input/nb-cassava-marcos/vgg16-cassava.pkl\"\nlearn = load_learner(model_path)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"data_dir = \"../input/cassava-leaf-disease-classification/test_images\"\n\ndf_predicao = pd.DataFrame(columns=['image_id','label'])\n\nfor _,_, imagens in os.walk(data_dir):\n    for imagem in imagens:\n        pred, clas, probs = learn.predict(data_dir+\"/\"+imagem)\n        df_predicao = df_predicao.append({'image_id':imagem, 'label': pred}, ignore_index=True)\n        \ndf_predicao.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}