-
Add LSTM and TCN models for predicting hysteretic curves · 6eb5ddfcJoaquín Irazábal González authored
- Implemented LSTM model with enhanced training features including mixed precision, early stopping, and validation via DataLoader. - Introduced a new 1D Temporal Convolutional Network (TCN) model for predicting hysteretic curves. - Added functionality for creating sliding windows from the dataset for both training and testing. - Integrated scaling for features and targets using StandardScaler. - Enhanced data handling with explicit cleanup and memory management for GPU. - Updated main function to include model training and prediction for both LSTM and TCN. - Saved predictions to CSV files for further analysis.
6eb5ddfc
| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| predict_hysteretic_curves.cpython-312.pyc | Loading commit data... |