Package com.aidenfavish.javaNeuralNetwork.Models

Class Model

java.lang.Object

com.aidenfavish.javaNeuralNetwork.Models.Model

public class Model
extends Object

This class is designed to hold layers and optimization methods to abstract the complex training and inference processes behind the scenes.

Constructor Summary

Constructors

Constructor	Description
Model(String path)
Model(String name, Optimizer optimizer)

Method Summary

Modifier and Type	Method	Description
void	addLayer(int index, LayerPass layer)
void	addLayer(LayerPass layer)
List<LayerPass>	getNetwork()
void	pass(float[][] dataX)
int	predict(float[] data)
void	save(String path)
String	toString()
void	train(int epochs, float[][] dataX, int[] dataY)
float	validate(float[][] dataX, int[] dataY)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Details

Model

public Model(String name,
 Optimizer optimizer)

Model

public Model(String path)

Method Details

addLayer

public void addLayer(LayerPass layer)

addLayer

public void addLayer(int index,
 LayerPass layer)

getNetwork

public List<LayerPass> getNetwork()

train

public void train(int epochs,
 float[][] dataX,
 int[] dataY)

validate

public float validate(float[][] dataX,
 int[] dataY)

predict

public int predict(float[] data)

pass

public void pass(float[][] dataX)

save

public void save(String path)

toString

public String toString()

Overrides:

toString in class Object