@webwriter/neural-network/src/types/model_conf.ts

Deep learning visualization for feed-forward networks with custom datasets, training and prediction.

import type * as tf from '@tensorflow/tfjs'

export interface ModelConf {
  model: tf.LayersModel
  loss: string
  metrics: string[]
  plottedMetrics: string[]
  isTraining: boolean
  totalEpochs: number
  actEpoch: number
  actBatch: number
  history: tf.Logs[]
  predictedValue: number | number[] // regression | classification
}