Skip to content

Commit

Permalink
some renaming
Browse files Browse the repository at this point in the history
  • Loading branch information
yasirusama61 committed Oct 9, 2024
1 parent a77de2a commit 467e029
Show file tree
Hide file tree
Showing 7 changed files with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ PSO is a population-based optimization algorithm inspired by the social behavior

### Algorithm: Particle Swarm Optimization

![PSO Algorithm](plots/pso_algorithm.png)
![PSO Algorithm](results/pso_algorithm.png)

## Evaluation Metrics

Expand All @@ -102,8 +102,8 @@ Results, including performance metrics and plots, are stored in the `results/` f

## Hyperparameter Optimization

![Batch Size Tuning](plots/batch_size_rmse.png)
![Block Number Tuning](plots/block_number_rmse.png)
![Batch Size Tuning](results/batch_size_rmse.png)
![Block Number Tuning](results/block_number_rmse.png)

During the development of the LSTM model, hyperparameter tuning was performed to achieve the optimal settings for better prediction accuracy. The table below summarizes the optimal hyperparameter values used in the final model:

Expand All @@ -124,10 +124,10 @@ During the development of the LSTM model, hyperparameter tuning was performed to

The figure below compares the predictions made by Univariate and Multivariate LSTM models against the actual heavy metal concentration.

![LSTM Comparison](plots/lstm_comparison.png)
![LSTM Comparison](results/lstm_comparison.png)

### Sensitivity Analysis Plot
![Sensitivity Analysis Plot](plots/sensitivity_analysis_plot.png)
![Sensitivity Analysis Plot](results/sensitivity_analysis_plot.png)

### ARIMA vs. LSTM Performance
In this project, both ARIMA (AutoRegressive Integrated Moving Average) and LSTM (Long Short-Term Memory) models were employed to forecast the heavy metal concentration in wastewater. The results were compared to evaluate the effectiveness of each model in time series forecasting:
Expand All @@ -138,7 +138,7 @@ In this project, both ARIMA (AutoRegressive Integrated Moving Average) and LSTM

- **Comparison Plot**: The plot below compares the predictions of ARIMA and LSTM against the actual heavy metal concentration values. Both models closely follow the actual trends, but LSTM shows a slight edge in accuracy, especially during rapid changes in concentration.

![ARIMA vs. LSTM Comparison](plots/arima_vs_lstm_comparison.png)
![ARIMA vs. LSTM Comparison](results/arima_vs_lstm_comparison.png)


### Conclusion
Expand Down
File renamed without changes
File renamed without changes
File renamed without changes
File renamed without changes
File renamed without changes
File renamed without changes

0 comments on commit 467e029

Please sign in to comment.