WenWeiTHU opened a new pull request, #9338:
URL: https://github.com/apache/iotdb/pull/9338
Add mlnode algoirthms
* support forecasting model: nbeats, dlinear
* factory method to create torch model: `create_forecast_model`
Usage:
```
from iotdb.mlnode.algorithm.model_factory import create_forecast_model
model, model_cfg = create_forecast_model(model_name='nbeats')
model, model_cfg = create_forecast_model(
model_name='dlinear'
input_len=96,
pred_len=96,
input_vars=7,
output_vars=7
)
# In fact, any kwargs can send to this fatory method
# and model_cfg will only have used kwargs for given type of model_name
#
# model, model_cfg = create_forecast_model(
# model_name='dlinear'
# input_len=96,
# pred_len=96,
# input_vars=7,
# output_vars=7,
# foo=0, bar='1' # useless kwargs
# )
# The returned model_cfg will not contain 'foo', 'bar'
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
