Home Reference Source Test
public class | source

EnsembleTrainerMixins

Extends:

BasePipelineClass → EnsembleTrainerMixins

The EnsembleTrainerMixins class is the mixins class for training ensamblable models

Method Summary

Public Methods
public

async ensembleTrain(numEpochs: Number , batchSize: Number , saveFileName: String ): Promise

Train model and save to storage under the given filename.

Public Methods

public async ensembleTrain(numEpochs: Number , batchSize: Number , saveFileName: String ): Promise source

Train model and save to storage under the given filename. parameters is reinited for each training

Params:

NameTypeAttributeDescription
numEpochs Number
batchSize Number
saveFileName String

Return:

Promise

Training losses