model.fit_generator()函數(shù)參數(shù)

fit_generator(self, generator, steps_per_epoch, epochs=1, verbose=1, callbacks=None, validation_data=None, validation_steps=None, class_weight=None, max_q_size=10, workers=1, pickle_safe=False, initial_epoch=0)

利用Python的生成器，逐個生成數(shù)據(jù)的batch并進行訓(xùn)練。生成器與模型將并行執(zhí)行以提高效率。例如，該函數(shù)允許我們在CPU上進行實時的數(shù)據(jù)提升，同時在GPU上進行模型訓(xùn)練

函數(shù)的參數(shù)是：

• generator：生成器函數(shù)，生成器的輸出應(yīng)該為：

  • 一個形如（inputs，targets）的tuple

  • 一個形如（inputs, targets,sample_weight）的tuple。所有的返回值都應(yīng)該包含相同數(shù)目的樣本。生成器將無限在數(shù)據(jù)集上循環(huán)。每個epoch以經(jīng)過模型的樣本數(shù)達到samples_per_epoch時，記一個epoch結(jié)束

• steps_per_epoch：整數(shù)，當(dāng)生成器返回steps_per_epoch次數(shù)據(jù)時計一個epoch結(jié)束，執(zhí)行下一個epoch

• epochs：整數(shù)，數(shù)據(jù)迭代的輪數(shù)

• verbose：日志顯示，0為不在標(biāo)準(zhǔn)輸出流輸出日志信息，1為輸出進度條記錄，2為每個epoch輸出一行記錄

• validation_data：具有以下三種形式之一

  • 生成驗證集的生成器

  • 一個形如（inputs,targets）的tuple

  • 一個形如（inputs,targets，sample_weights）的tuple

• validation_steps: 當(dāng)validation_data為生成器時，本參數(shù)指定驗證集的生成器返回次數(shù)

• class_weight：規(guī)定類別權(quán)重的字典，將類別映射為權(quán)重，常用于處理樣本不均衡問題。

• sample_weight：權(quán)值的numpy array，用于在訓(xùn)練時調(diào)整損失函數(shù)（僅用于訓(xùn)練）。可以傳遞一個1D的與樣本等長的向量用于對樣本進行1對1的加權(quán)，或者在面對時序數(shù)據(jù)時，傳遞一個的形式為（samples，sequence_length）的矩陣來為每個時間步上的樣本賦不同的權(quán)。這種情況下請確定在編譯模型時添加了sample_weight_mode='temporal'。

• workers：最大進程數(shù)

• max_q_size：生成器隊列的最大容量

• pickle_safe: 若為真，則使用基于進程的線程。由于該實現(xiàn)依賴多進程，不能傳遞non picklable（無法被pickle序列化）的參數(shù)到生成器中，因為無法輕易將它們傳入子進程中。

• initial_epoch: 從該參數(shù)指定的epoch開始訓(xùn)練，在繼續(xù)之前的訓(xùn)練時有用。

?

generator生成器

def generate_arrays_from_file(path):while 1:f = open(path)for line in f:# create Numpy arrays of input data# and labels, from each line in the filex, y = process_line(line)yield (x, y)f.close()model.fit_generator(generate_arrays_from_file('/my_file.txt'),samples_per_epoch=10000, epochs=10)

總結(jié)

以上是生活随笔為你收集整理的2020-12-08 tensorflow model.fit_generator()函数参数的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。