mirror of
https://github.com/TheAlgorithms/Python.git
synced 2025-04-01 11:26:43 +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
|
||||
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.hidden_layer_size: int = hidden_layer_size
|
||||
self.training_epochs: int = training_epochs
|
||||
|
Loading…
x
Reference in New Issue
Block a user