记录训练任务详细信息的keras扩展(回调)
Keras-Experiment-Recorder的Python项目详细描述
Keras实验记录器
这是一组帮助存储 在firebase后端进行实验(如模型信息、训练结果等)。
先决条件
您将需要一个firebase应用程序的服务帐户。
用法
fromkeras_experiment_recorderimportExperimentInfo,FirebaseInfofromkeras_experiment_recorderimportExperimentRecorder# below three lines setup the instance of ExperimentRecorderfb_info=FirebaseInfo(bucket_name=bucket_name,credential_path=credential_path)experiment_info=ExperimentInfo(name='first exp',fb_info=fb_info)recorder=ExperimentRecorder(experiment_info)callbacks=[recorder]history=model.fit(x_train,y_train,batch_size=batch_size,epochs=epochs,verbose=1,callbacks=callbacks,# pass the callbacksvalidation_split=0.1)
运行此存储库中包含的示例
python examples/simple.py --bucket-name keras-experiment-recorder.appspot.com --credential-path ~/Desktop/Dev/keys/keras-experimen t-recorder-firebase.json