Mirrors_Framework/RBF-aPID-Controller
1
0
Fork 0
mirror of https://github.com/WallabyLester/RBF-aPID-Controller.git synced 2026-05-10 04:58:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Moved input_dim parameter to class constructor.

Browse Source
This commit is contained in:
Andru Liu
2024-10-07 19:57:40 -07:00
parent 9ac342f003
commit cc6ff7ca80
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
RBF_tf.py
+8 -4
View File
@@ -17,7 +17,7 @@ class RBFLayer(tf.keras.layers.Layer):
call(inputs):
TF call method to implement forward pass.
"""
def __init__(self, n_centers, input_dim=3):
def __init__(self, n_centers, input_dim):
""" Constructs RBF centers and standard deviations as trainable weights.
Parameters
@@ -25,7 +25,7 @@ class RBFLayer(tf.keras.layers.Layer):
n_centers : int
The number of RBF centers.
input_dim : int
The dimensions of the RBF centers. Default of 3 for Kp, Ki, Kd
The dimensions of the RBF centers.
"""
super().__init__()
self.n_centers = n_centers
@@ -66,22 +66,26 @@ class RBFAdaptiveModel(tf.keras.Model):
----------
n_centers : int
The number of RBF centers.
imput_dim : int
The dimensions of the RBF centers.
Methods
-------
call(inputs):
TF call method to implement forward pass of model.
"""
def __init__(self, n_centers):
def __init__(self, n_centers, input_dim=3):
""" Constructs RBF and output layers.
Parameters
----------
n_centers : int
The number of RBF centers.
input_dim : int
The dimensions of the RBF centers. Default of 3 for Kp, Ki, Kd
"""
super().__init__()
self.rbf_layer = RBFLayer(n_centers)
self.rbf_layer = RBFLayer(n_centers, input_dim)
self.output_layer = tf.keras.layers.Dense(1)
def call(self, inputs):