search

zhouhaoyi / Informer2020

The GitHub repository for the paper "Informer" accepted by AAAI 2021.
Apache License 2.0
5.24k stars 1.1k forks source link

new release #623

Closed cloner174 closed 3 months ago

cloner174 commented 3 months ago

Hi, This is a enhanced version of the original model releases in 2021. Here I try to simplify the process and add some necessary (on my point of view of course) to it! What:

Enhanced the Exp_Informer class by adding more flexibility and options.
Introduced several improvements to better support custom data inputs and various optimization methods.

Why:

To streamline the data processing workflow.
To offer more customization options for users with different requirements.

How:

Added a new parameter direct_data for direct data input without reading from files.
Introduced new attributes: train_losses, test_losses, actual_test_values, and predicted_test_values for better tracking of training and testing metrics.
Enhanced _get_data method to handle custom data and various shuffling options.
Expanded _select_optimizer method to support a variety of optimizers.
Improved _select_criterion method to support multiple loss functions.
Updated training, validation, and testing methods to utilize new parameters and attributes.