AutoKeras训练mnist效果分析

搜索时间为24h,代码:

from autokeras.image.image_supervised import load_image_dataset, ImageClassifier
from keras.models import load_model
from keras.utils import plot_model
from autokeras.utils import pickle_from_file
from keras.preprocessing.image import load_img, img_to_array
import numpy as np
import time
from keras.datasets import mnist

(train_data, train_labels), (test_data, test_labels) = mnist.load_data()
train_data = train_data.reshape(train_data.shape + (1,))
train_labels = train_labels
test_data = test_data.reshape(test_data.shape + (1,))
test_labels = test_labels

MODEL_DIR = './my_model.h5'
MODEL_PNG = './model.png'

# 数据进行格式转换
train_data = train_data.astype('float32')
test_data = test_data.astype('float32')
print("train data shape:", train_data.shape)
# 使用图片识别器
clf = ImageClassifier(path="./automodels/",verbose=True, augment=False)
# 给其训练数据和标签,训练的最长时间可以设定,假设为1分钟,autokers会不断找寻最优的网络模型
clf.fit(train_data, train_labels, time_limit=24*60*60)
# 找到最优模型后,再最后进行一次训练和验证
#clf.final_fit(train_data, train_labels, test_data, test_labels, retrain=True)
# 给出评估结果
y = clf.evaluate(test_data, test_labels)
print("evaluate:", y)
# 导出我们生成的模型
clf.export_autokeras_model(MODEL_DIR)
# 加载模型
model = pickle_from_file(MODEL_DIR)

结果:

evaluate: 0.9935

模型大小:

45M
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 首页 资讯 文章 资源 小组 相亲 登录 注册 首页 最新文章 IT 职场 前端 后端 移动端 数据库 运维 其他...
    Helen_Cat阅读 3,955评论 1 10
  • Swift1> Swift和OC的区别1.1> Swift没有地址/指针的概念1.2> 泛型1.3> 类型严谨 对...
    cosWriter阅读 11,161评论 1 32
  • 【日精打卡第77天】 姓 名: 夏 婷 公 司:无锡镁钛铒金属制品有限公司 六项精进第423期 反省二组 【知~学...
    xia芊芊阅读 157评论 0 0
  • 有时候,我们会觉得自己很失败,尤其是当别人获得荣誉的时候,心里很羡慕,也很难受为什么不是自己。但是一觉醒来,他们的...
    一条爱飞的鱼阅读 315评论 0 2
  • 2017年9月27日 星期三 多云 最近一段时间加班有点多,甚至有点儿晚,没能有多余的时间...
    千雨轩阅读 233评论 1 4