Jun 15, 2024 · Jun 16, 2024 · Jun 16, 2024 · Jun 16, 2024 · Jun 16, 2024 · Jun 16, 2024
diff --git a/src/neural-network.ts b/src/neural-network.ts
  constructor(
    options: Partial<INeuralNetworkOptions & INeuralNetworkTrainOptions> = {}
  ) {
    this.options = { ...this.options, ...options };
    this.options.binaryThresh = options.binaryThresh ?? 0.5;
    this.options.hiddenLayers = options.hiddenLayers ?? [];
    this.options.inputSize = options.inputSize ?? 1;
    this.options.loss = options.loss ?? loss;
    this.options.outputSize = options.outputSize ?? options.inputSize ?? 1;
    this.options.ramSize = options.ramSize ?? 1;
    this.updateTrainingOptions(options);

    const { inputSize, hiddenLayers, outputSize } = this.options;
diff --git a/src/utilities/to-svg.ts b/src/utilities/to-svg.ts
  // Get network size array for NeuralNetwork or NeuralNetworkGPU
  let sizes: number[] = [];
  if (net instanceof NeuralNetwork || net instanceof NeuralNetworkGPU) {
    sizes = getNeuralNetworkSizes(net);
    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
    // @ts-expect-error
    sizes = getNeuralNetworkSizes((net as unknown) as NeuralNetwork);
  }
  // get network size for Recurrent
  else if (net instanceof Recurrent) {
Original file line number	Diff line number	Diff line change
Expand Up		@@ -323,7 +323,12 @@ export class NeuralNetwork<
		constructor(
		options: Partial<INeuralNetworkOptions & INeuralNetworkTrainOptions> = {}
		) {
		this.options = { ...this.options, ...options };
		this.options.binaryThresh = options.binaryThresh ?? 0.5;
		this.options.hiddenLayers = options.hiddenLayers ?? [];
		this.options.inputSize = options.inputSize ?? 1;
		this.options.loss = options.loss ?? loss;
		this.options.outputSize = options.outputSize ?? options.inputSize ?? 1;
		this.options.ramSize = options.ramSize ?? 1;
		this.updateTrainingOptions(options);

		const { inputSize, hiddenLayers, outputSize } = this.options;
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -467,7 +467,9 @@ export function toSVG<
		// Get network size array for NeuralNetwork or NeuralNetworkGPU
		let sizes: number[] = [];
		if (net instanceof NeuralNetwork \|\| net instanceof NeuralNetworkGPU) {
		sizes = getNeuralNetworkSizes(net);
		// eslint-disable-next-line @typescript-eslint/ban-ts-comment
		// @ts-expect-error
		sizes = getNeuralNetworkSizes((net as unknown) as NeuralNetwork);
		}
		// get network size for Recurrent
		else if (net instanceof Recurrent) {
Expand Down