From 08d3ac7ef821e42e428d8250c279fb00c8da553e Mon Sep 17 00:00:00 2001 From: robcaulk Date: Fri, 29 Jul 2022 08:49:35 +0200 Subject: [PATCH] add keras and conv_width to schema and documentation --- docs/freqai.md | 3 +++ freqtrade/constants.py | 2 ++ 2 files changed, 5 insertions(+) diff --git a/docs/freqai.md b/docs/freqai.md index 4060b5394..3c04d5b31 100644 --- a/docs/freqai.md +++ b/docs/freqai.md @@ -110,6 +110,9 @@ Mandatory parameters are marked as **Required**, which means that they are requi | `n_estimators` | A common parameter among regressors which sets the number of boosted trees to fit
**Datatype:** integer. | `learning_rate` | A common parameter among regressors which sets the boosting learning rate.
**Datatype:** float. | `n_jobs`, `thread_count`, `task_type` | Different libraries use different parameter names to control the number of threads used for parallel processing or whether or not it is a `task_type` of `gpu` or `cpu`.
**Datatype:** float. +| | **Extraneous parameters** +| `keras` | If your model makes use of keras (typical of Tensorflow based prediction models), activate this flag so that the model save/loading follows keras standards. Default value `false`
**Datatype:** boolean. +| `conv_width` | The width of a convolutional neural network input tensor. This replaces the need for `shift` by feeding in historical data points as the second dimension of the tensor. Technically, this parameter can also be used for regressors, but it only adds computational overhead and does not change the model training/prediction. Default value, 2
**Datatype:** integer. ### Important FreqAI dataframe key patterns diff --git a/freqtrade/constants.py b/freqtrade/constants.py index 90c87809b..0134dafc0 100644 --- a/freqtrade/constants.py +++ b/freqtrade/constants.py @@ -481,6 +481,8 @@ CONF_SCHEMA = { "freqai": { "type": "object", "properties": { + "keras": {"type": "boolean", "default": False}, + "conv_width": {"type": "integer", "default": 2}, "train_period_days": {"type": "integer", "default": 0}, "backtest_period_days": {"type": "float", "default": 7}, "identifier": {"type": "str", "default": "example"},