mirror of
https://github.com/TheAlgorithms/Python.git
synced 2025-04-02 20:06:50 +00:00
added type hints in lstm init
This commit is contained in:
parent
750c9f6fc8
commit
88ac16b1e6
@ -36,6 +36,28 @@ class LongShortTermMemory:
|
|||||||
>>> lstm.learning_rate
|
>>> lstm.learning_rate
|
||||||
0.01
|
0.01
|
||||||
"""
|
"""
|
||||||
|
self.input_data: str
|
||||||
|
self.hidden_layer_size: int
|
||||||
|
self.training_epochs: int
|
||||||
|
self.learning_rate: float
|
||||||
|
self.unique_chars: set[str]
|
||||||
|
self.data_length: int
|
||||||
|
self.vocabulary_size: int
|
||||||
|
self.char_to_index: dict[str, int]
|
||||||
|
self.index_to_char: dict[int, str]
|
||||||
|
self.input_sequence: str
|
||||||
|
self.target_sequence: str
|
||||||
|
self.random_generator: Generator
|
||||||
|
self.combined_inputs: dict[int, np.ndarray]
|
||||||
|
self.hidden_states: dict[int, np.ndarray]
|
||||||
|
self.cell_states: dict[int, np.ndarray]
|
||||||
|
self.forget_gate_activations: dict[int, np.ndarray]
|
||||||
|
self.input_gate_activations: dict[int, np.ndarray]
|
||||||
|
self.cell_state_candidates: dict[int, np.ndarray]
|
||||||
|
self.output_gate_activations: dict[int, np.ndarray]
|
||||||
|
self.network_outputs: dict[int, np.ndarray]
|
||||||
|
|
||||||
|
|
||||||
self.input_data: str = input_data.lower()
|
self.input_data: str = input_data.lower()
|
||||||
self.hidden_layer_size: int = hidden_layer_size
|
self.hidden_layer_size: int = hidden_layer_size
|
||||||
self.training_epochs: int = training_epochs
|
self.training_epochs: int = training_epochs
|
||||||
|
Loading…
x
Reference in New Issue
Block a user