rename base_master_nn to nn_base

henrysky · Sep 7, 2024 · 9c0ebc4 · 9c0ebc4
1 parent acd235c
commit 9c0ebc4
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/docs/source/neuralnets/basic_usage.rst b/docs/source/neuralnets/basic_usage.rst
@@ -18,7 +18,7 @@ relies relies on two major component, `Normalizer` and `GeneratorMaster`
     ├── Bayesian_DataGenerator
     └── CVAE_DataGenerator
 
-    NeuralNetBase (astroNN.models.base_master_nn.NeuralNetBase)
+    NeuralNetBase (astroNN.models.nn_base.NeuralNetBase)
     ├── CNNBase
     │   ├── ApogeeCNN
     │   ├── StarNet2017
@@ -38,10 +38,10 @@ relies relies on two major component, `Normalizer` and `GeneratorMaster`
 NeuralNetBase Class API
 ------------------------------
 
-All astroNN Neural Nets classes inherited from this  ``astroNN.models.base_master_nn.NeuralNetBase`` and thus methods
+All astroNN Neural Nets classes inherited from this  ``astroNN.models.nn_base.NeuralNetBase`` and thus methods
 of this class is shared across all astroNN Neural Nets classes.
 
-.. autoclass:: astroNN.models.base_master_nn.NeuralNetBase
+.. autoclass:: astroNN.models.nn_base.NeuralNetBase
     :members:
 
 -----------

diff --git a/src/astroNN/models/base_bayesian_cnn.py b/src/astroNN/models/base_bayesian_cnn.py
@@ -12,7 +12,7 @@
     backend_framework,
 )
 from astroNN.config import _astroNN_MODEL_NAME
-from astroNN.models.base_master_nn import NeuralNetBase
+from astroNN.models.nn_base import NeuralNetBase
 from astroNN.nn.callbacks import VirutalCSVLogger
 from astroNN.nn.layers import FastMCInference
 from astroNN.nn.losses import (

diff --git a/src/astroNN/models/base_cnn.py b/src/astroNN/models/base_cnn.py
@@ -8,7 +8,7 @@
 import keras
 from astroNN.config import MULTIPROCESS_FLAG
 from astroNN.config import _astroNN_MODEL_NAME
-from astroNN.models.base_master_nn import NeuralNetBase
+from astroNN.models.nn_base import NeuralNetBase
 from astroNN.nn.callbacks import VirutalCSVLogger
 from astroNN.nn.losses import categorical_crossentropy, binary_crossentropy
 from astroNN.nn.losses import mean_squared_error, mean_absolute_error, mean_error

diff --git a/src/astroNN/models/base_vae.py b/src/astroNN/models/base_vae.py
@@ -9,7 +9,7 @@
 from astroNN.config import MULTIPROCESS_FLAG, _KERAS_BACKEND, backend_framework
 from astroNN.config import _astroNN_MODEL_NAME
 from astroNN.datasets import H5Loader
-from astroNN.models.base_master_nn import NeuralNetBase
+from astroNN.models.nn_base import NeuralNetBase
 from astroNN.nn.callbacks import VirutalCSVLogger
 from astroNN.nn.losses import (
     mean_squared_reconstruction_error,

diff --git a/src/astroNN/models/base_master_nn.py → src/astroNN/models/nn_base.py b/src/astroNN/models/base_master_nn.py → src/astroNN/models/nn_base.py
@@ -1,5 +1,5 @@
 ###############################################################################
-#   base_master_nn.py: top-level class for a neural network
+#   nn_base.py: top-level class for a neural network
 ###############################################################################
 import os
 import sys