新手上路。
这是我正在处理的数据集:https://www.kaggle.com/arpitjain007/game-of-deep-learning-ship-datasets
我使用的是fastai,我已经成功构建了模型,但我不知道如何使用“test.csv”文件进行测试。
这是我的代码:
from fastai import *
from fastai.vision import *
path = '../input/train'
path = Path(path)
path.ls()
df = pd.read_csv(path/'train.csv')
data = ImageDataBunch.from_df('../input/train/images', df, ds_tfms=get_transforms(), size=224, bs=64 ).normalize(imagenet_stats)
learn = cnn_learner(data, models.resnet50, metrics=accuracy, model_dir='/kaggle/working/models')
learn.fit_one_cycle(5)
df_test = pd.read_csv('../input/test_ApKoW4T.csv')
我不知道如何使用测试数据框来进行预测。