TfModelSummary

data class TfModelSummary(type: String, name: String?, layersSummaries: List<LayerSummary>, trainableParamsCount: Long, frozenParamsCount: Long) : ModelSummary

The common information about model.

Constructors

TfModelSummary

fun TfModelSummary(type: String, name: String?, layersSummaries: List<LayerSummary>, trainableParamsCount: Long, frozenParamsCount: Long)

Functions

customFormat

fun customFormat(layerNameColumnName: String = "Layer (type) ", outputShapeColumnName: String = "Output Shape ", paramsCountColumnName: String = "Param # ", connectedToColumnName: String = "Connected to ", columnSeparator: String = " ", lineSeparatorSymbol: Char = '_', thickLineSeparatorSymbol: Char = '=', withConnectionsColumn: Boolean = layersSummaries.any { it.inboundLayers.size > 1 }): List<String>

Formats model summary

format