自動機器學習框架之三_Auto-Keras

?對于訓練深度學習,設計神經網絡結構是其中技術含高最高的任務,優(yōu)秀的網絡架構往往依賴建構模型的經驗炫狱,專業(yè)領域知識藻懒,以及大量的算力試錯。實際應用中往往基于類似功能的神經網絡微調生成新的網絡結構视译。

?Auto-Keras是一個離線使用的開源庫嬉荆,用于構建神經網絡結構和搜索超參數(shù),支持RNN酷含,CNN神經網絡鄙早,它使用了高效神經網絡搜索ENAS,利用遷移學習的原理將在前面任務中學到的權值應用于后期的模型中椅亚,效率相對較高限番。除了支持keras,它還提供TensorFlow他PyTorch的版本呀舔。

1. 安裝

?由于需要把輸出的神經網絡結構保存成圖片弥虐,使用了pydot和graphviz圖像工具支持,auto-keras支持torch等多種工具支持媚赖,因此霜瘪,安裝時會下載大量依賴軟件。使用以下方法安裝auto-keras:

$ apt install graphviz 
$ pip3 install pydot
$ pip3 install autokeras

?使用以下方法下載源碼:

$ git clone https://github.com/jhfjhfj1/autokeras

2. 舉例

?本例中使用了mnist數(shù)據集惧磺,它是一個入門級的圖像識別數(shù)據集粥庄,用于訓練手寫數(shù)字識別模型,例程自動下載訓練數(shù)據豺妓,然后創(chuàng)建了圖片分類器,訓練時間設置為10分鐘布讹,模型在測試集上的正確率為99.21%琳拭。建議使用帶GPU的機器訓練模型,它比使用CPU訓練速度快幾十倍描验。

from keras.datasets import mnist
from autokeras import ImageClassifier
from autokeras.constant import Constant
import autokeras
from keras.utils import plot_model
    
if __name__ == '__main__':
    (x_train, y_train), (x_test, y_test) = mnist.load_data()
    x_train = x_train.reshape(x_train.shape + (1,))
    x_test = x_test.reshape(x_test.shape + (1,))
    clf = ImageClassifier(verbose=True, augment=False)
    clf.fit(x_train, y_train, time_limit=10 * 60)
    clf.final_fit(x_train, y_train, x_test, y_test, retrain=True)
    y = clf.evaluate(x_test, y_test)
    print(y * 100)
    clf.export_keras_model('model.h5')
    plot_model(clf, to_file='model.png')
# 返回值: 99.21
# 謝彥的技術博客

?上述程序在我的機器上運行后訓練出了17層網絡白嘁,其中包括dropout層,池化層膘流,卷積層絮缅,全連接層等,程序以圖片方式將描述信息保存在model.png中呼股,下面截取了圖片中的部分層作為示例耕魄,如下圖所示:


最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市彭谁,隨后出現(xiàn)的幾起案子吸奴,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 211,743評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件则奥,死亡現(xiàn)場離奇詭異考润,居然都是意外死亡,警方通過查閱死者的電腦和手機读处,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,296評論 3 385
  • 文/潘曉璐 我一進店門糊治,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人罚舱,你說我怎么就攤上這事井辜。” “怎么了馆匿?”我有些...
    開封第一講書人閱讀 157,285評論 0 348
  • 文/不壞的土叔 我叫張陵抑胎,是天一觀的道長。 經常有香客問我渐北,道長阿逃,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 56,485評論 1 283
  • 正文 為了忘掉前任赃蛛,我火速辦了婚禮恃锉,結果婚禮上,老公的妹妹穿的比我還像新娘呕臂。我一直安慰自己破托,他們只是感情好,可當我...
    茶點故事閱讀 65,581評論 6 386
  • 文/花漫 我一把揭開白布歧蒋。 她就那樣靜靜地躺著土砂,像睡著了一般。 火紅的嫁衣襯著肌膚如雪谜洽。 梳的紋絲不亂的頭發(fā)上萝映,一...
    開封第一講書人閱讀 49,821評論 1 290
  • 那天,我揣著相機與錄音阐虚,去河邊找鬼序臂。 笑死,一個胖子當著我的面吹牛实束,可吹牛的內容都是我干的奥秆。 我是一名探鬼主播,決...
    沈念sama閱讀 38,960評論 3 408
  • 文/蒼蘭香墨 我猛地睜開眼咸灿,長吁一口氣:“原來是場噩夢啊……” “哼构订!你這毒婦竟也來了?” 一聲冷哼從身側響起避矢,我...
    開封第一講書人閱讀 37,719評論 0 266
  • 序言:老撾萬榮一對情侶失蹤鲫咽,失蹤者是張志新(化名)和其女友劉穎签赃,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體分尸,經...
    沈念sama閱讀 44,186評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡锦聊,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,516評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了箩绍。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片孔庭。...
    茶點故事閱讀 38,650評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖材蛛,靈堂內的尸體忽然破棺而出圆到,到底是詐尸還是另有隱情,我是刑警寧澤卑吭,帶...
    沈念sama閱讀 34,329評論 4 330
  • 正文 年R本政府宣布芽淡,位于F島的核電站,受9級特大地震影響豆赏,放射性物質發(fā)生泄漏挣菲。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,936評論 3 313
  • 文/蒙蒙 一掷邦、第九天 我趴在偏房一處隱蔽的房頂上張望白胀。 院中可真熱鬧,春花似錦抚岗、人聲如沸或杠。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,757評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽向抢。三九已至,卻和暖如春胚委,著一層夾襖步出監(jiān)牢的瞬間挟鸠,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,991評論 1 266
  • 我被黑心中介騙來泰國打工篷扩, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人茉盏。 一個月前我還...
    沈念sama閱讀 46,370評論 2 360
  • 正文 我出身青樓鉴未,卻偏偏與公主長得像,于是被迫代替她去往敵國和親鸠姨。 傳聞我的和親對象是個殘疾皇子铜秆,可洞房花燭夜當晚...
    茶點故事閱讀 43,527評論 2 349