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

Public Member Functions

Public Attributes