हर युग के बाद एक मॉडल को बचाने के लिए केरस कॉलबैक उदाहरण?


17

क्या हर काल के बाद किसी मॉडल को बचाने के लिए कॉलबैक का उपयोग करके केरेस का एक सीधा उदाहरण पोस्ट कर सकते हैं? मैं वजन बचाने के उदाहरण पा सकता हूं, लेकिन मैं चाहता हूं कि हर प्रशिक्षण काल ​​के बाद पूरी तरह से कामकाजी मॉडल को बचाने में सक्षम हो।

जवाबों:


22

केरस कॉलबैक में 'save_weights_only' को गलत पर सेट करने से 'ModelCheckpoint' पूर्ण मॉडल को बचाएगा; ऊपर दिए गए लिंक से लिया गया यह उदाहरण, प्रदर्शन की परवाह किए बिना, प्रत्येक पूर्ण युग को बचाएगा।

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

यहां कुछ और उदाहरण पाए जाते हैं , जिसमें केवल बेहतर मॉडल को सहेजना और सहेजे गए मॉडल को लोड करना शामिल है।


10

अपने फाइलपथ में युगांतर चर को शामिल करना सुनिश्चित करें। अन्यथा आपके सहेजे गए मॉडल को हर काल के बाद बदल दिया जाएगा।

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

अधिक उदाहरणों के लिए, यहां देखें


2
साइट पर आपका स्वागत है! और धन्यवाद, मैं जवाब के अलावा उस सराहना करता हूं
I_Play_With_Data
हमारी साइट का प्रयोग करके, आप स्वीकार करते हैं कि आपने हमारी Cookie Policy और निजता नीति को पढ़ और समझा लिया है।
Licensed under cc by-sa 3.0 with attribution required.