pip install tensorflow matplotlib numpy我們還需要下載MNIST數(shù)據(jù)集,可以使用TensorFlow的內(nèi)置函數(shù)進(jìn)行下載:
python import tensorflow as tf mnist = tf.keras.datasets.mnist (x_train, y_train), (x_test, y_test) = mnist.load_data()這將下載并加載MNIST數(shù)據(jù)集,將訓(xùn)練集和測試集分別存儲在`x_train`、`y_train`和`x_test`、`y_test`中。 ## 構(gòu)建模型 接下來,我們將構(gòu)建一個(gè)包含3個(gè)全連接層的神經(jīng)網(wǎng)絡(luò)。首先,我們需要將輸入數(shù)據(jù)展平為1維向量,然后將其輸入到第一個(gè)全連接層中。每個(gè)全連接層后面都跟著一個(gè)ReLU激活函數(shù)和一個(gè)Dropout層,以避免過擬合。
python model = tf.keras.models.Sequential([ tf.keras.layers.Flatten(input_shape=(28, 28)), tf.keras.layers.Dense(128, activation="relu"), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(64, activation="relu"), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(10) ])最后一層不使用激活函數(shù),因?yàn)槲覀儗⒃谟?xùn)練期間使用softmax來計(jì)算輸出。 ## 訓(xùn)練模型 接下來,我們需要編譯并訓(xùn)練我們的模型。我們將使用adam優(yōu)化器和交叉熵?fù)p失函數(shù)。
python model.compile(optimizer="adam", loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True), metrics=["accuracy"]) model.fit(x_train, y_train, epochs=10)這將對模型進(jìn)行10個(gè)epoch的訓(xùn)練。我們可以使用測試集來評估模型的性能:
python test_loss, test_acc = model.evaluate(x_test, y_test, verbose=2) print("## 預(yù)測 最后,我們可以使用訓(xùn)練好的模型來預(yù)測新的手寫數(shù)字。我們可以使用`matplotlib`庫來顯示圖像,并使用`argmax`函數(shù)來查找模型輸出中最大的元素的索引。 ```python import matplotlib.pyplot as plt import numpy as np # 顯示圖像 plt.imshow(x_test[0], cmap=plt.cm.binary) plt.show() # 預(yù)測結(jié)果 predictions = model.predict(np.array([x_test[0]])) print(np.argmax(predictions[0]))
Test accuracy:", test_acc)
文章版權(quán)歸作者所有,未經(jīng)允許請勿轉(zhuǎn)載,若此文章存在違規(guī)行為,您可以聯(lián)系管理員刪除。
轉(zhuǎn)載請注明本文地址:http://www.ezyhdfw.cn/yun/130754.html
摘要:近日,該論文的一作終于在上公開了該論文中的代碼。該項(xiàng)目上線天便獲得了個(gè),并被了次。 當(dāng)前的深度學(xué)習(xí)理論是由Geoffrey Hinton大神在2007年確立起來的,但是如今他卻認(rèn)為,CNN的特征提取層與次抽樣層交叉存取,將相同類型的相鄰特征檢測器的輸出匯集到一起是大有問題的。去年9月,在多倫多接受媒體采訪時(shí),Hinton大神斷然宣稱要放棄反向傳播,讓整個(gè)人工智能從頭再造。10月,人們關(guān)注已久...
摘要:概覽概覽產(chǎn)品簡介基礎(chǔ)知識產(chǎn)品優(yōu)勢機(jī)制產(chǎn)品架構(gòu)設(shè)計(jì)原理彈性擴(kuò)縮容機(jī)制開發(fā)綜述服務(wù)請求方式開源鏡像開源案例學(xué)習(xí)視頻產(chǎn)品定價(jià)快速上手快速上手案例介紹環(huán)境準(zhǔn)備在線服務(wù)代碼簡介 概覽產(chǎn)品簡介UAI-Inference基礎(chǔ)知識產(chǎn)品優(yōu)勢Hot-Standby機(jī)制產(chǎn)品架構(gòu)設(shè)計(jì)原理彈性擴(kuò)縮容機(jī)制開發(fā)綜述服務(wù)請求方式開源Docker鏡像開源案例學(xué)習(xí)視頻產(chǎn)品定價(jià)快速上手快速上手(TF-Mnist案例)MNIST ...
摘要:前言本文基于官網(wǎng)的寫成。是自帶的一個(gè)可視化工具,是其中的一個(gè)功能,用于在二維或三維空間對高維數(shù)據(jù)進(jìn)行探索。本文使用數(shù)據(jù)講解的使用方法。 前言 本文基于TensorFlow官網(wǎng)的How-Tos寫成。 TensorBoard是TensorFlow自帶的一個(gè)可視化工具,Embeddings是其中的一個(gè)功能,用于在二維或三維空間對高維數(shù)據(jù)進(jìn)行探索。 An embedding is a map ...
閱讀 2909·2023-04-25 20:06
閱讀 1532·2021-08-26 14:15
閱讀 2295·2021-08-12 13:27
閱讀 1841·2019-08-30 15:55
閱讀 3528·2019-08-30 13:20
閱讀 2890·2019-08-29 15:12
閱讀 3388·2019-08-29 15:06
閱讀 2925·2019-08-29 14:13