StatsCallback

Inheritance diagram for StatsCallback:

Collaboration diagram for StatsCallback:

Public Member Functions
def	__init__ (self, args, data, mdn, metrics=[mdsa, sspb], folder='Results_gpu')
def	on_train_batch_end (self, batch, logs=None)

Public Attributes
	start_time
	args
	data
	mdn
	metrics
	folder

Detailed Description

Save performance statistics as the model is trained 

Definition at line 31 of file callbacks.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			args,
			data,
			mdn,
			metrics = [mdsa, sspb],
			folder = 'Results_gpu'
	)

Definition at line 34 of file callbacks.py.

Member Function Documentation

on_train_batch_end()

def on_train_batch_end	(		self,
			batch,
			logs = None
	)

Definition at line 44 of file callbacks.py.

Member Data Documentation

args

args

Definition at line 38 of file callbacks.py.

data

data

Definition at line 39 of file callbacks.py.

folder

folder

Definition at line 42 of file callbacks.py.

mdn

mdn

Definition at line 40 of file callbacks.py.

metrics

metrics

Definition at line 41 of file callbacks.py.

start_time

start_time

Definition at line 37 of file callbacks.py.

---

The documentation for this class was generated from the following file:

• /gfs-oceanweb/web/ocssw/ocssw_src/src/aquaverse/MDN/model/callbacks.py