From c5c84383ba5fac5e4630e6a940e5284be1f90326 Mon Sep 17 00:00:00 2001
From: voidvoxel <160202125+voidvoxel@users.noreply.github.com>
Date: Mon, 17 Jun 2024 15:13:15 -0400
Subject: [PATCH] TypeScript hates me today

---
 src/autoencoder.ts                           | 9 +++------
 src/errors/untrained-neural-network-error.ts | 8 +++++++-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/autoencoder.ts b/src/autoencoder.ts
index bebed4dc..71d6b386 100644
--- a/src/autoencoder.ts
+++ b/src/autoencoder.ts
@@ -35,15 +35,12 @@ export class AutoencoderGPU<DecodedData extends INeuralNetworkData, EncodedData
     // Create default options for the autoencoder.
     options ??= {};
 
-    // Inherit the binary threshold of the parent autoencoder.
-    options.binaryThresh = options.binaryThresh;
-    // Inherit the hidden layers of the parent autoencoder.
-    options.hiddenLayers = options.hiddenLayers;
-
     const decodedSize = options.inputSize ?? options.outputSize ?? 1;
 
     // Define the denoiser subnet's input and output sizes.
-    if (decodedSize) options.inputSize = options.outputSize = decodedSize;
+    options.inputSize = options.outputSize = decodedSize;
+
+    options.hiddenLayers ??= [ Math.round(decodedSize * 0.66) ];
 
     options.loss ??= loss;
 
diff --git a/src/errors/untrained-neural-network-error.ts b/src/errors/untrained-neural-network-error.ts
index bd7722f3..cf643e78 100644
--- a/src/errors/untrained-neural-network-error.ts
+++ b/src/errors/untrained-neural-network-error.ts
@@ -1,6 +1,12 @@
+import { NeuralNetwork } from "../neural-network";
+
+interface IErrorableNeuralNetwork {
+  constructor: Function;
+}
+
 export class UntrainedNeuralNetworkError extends Error {
   constructor (
-    neuralNetwork: object
+    neuralNetwork: IErrorableNeuralNetwork
   ) {
     super(`Cannot run a ${neuralNetwork.constructor.name} before it is trained.`);
   }