freqtrade_origin/freqtrade/templates/sample_hyperopt.py

175 lines
7.0 KiB
Python
Raw Normal View History

2018-11-07 18:46:04 +00:00
# pragma pylint: disable=missing-docstring, invalid-name, pointless-string-statement
2020-09-28 17:39:41 +00:00
# isort: skip_file
2018-11-07 18:46:04 +00:00
2019-11-21 05:49:16 +00:00
# --- Do not remove these libs ---
2018-11-07 18:46:04 +00:00
from functools import reduce
from typing import Any, Callable, Dict, List
2018-11-07 18:46:04 +00:00
import numpy as np # noqa
2019-11-21 05:49:16 +00:00
import pandas as pd # noqa
from pandas import DataFrame
from skopt.space import Categorical, Dimension, Integer, Real # noqa
2018-11-07 18:46:04 +00:00
2018-11-20 16:40:45 +00:00
from freqtrade.optimize.hyperopt_interface import IHyperOpt
2018-11-07 18:46:04 +00:00
2019-11-21 05:49:16 +00:00
# --------------------------------
# Add your lib to import here
import talib.abstract as ta # noqa
import freqtrade.vendor.qtpylib.indicators as qtpylib
2018-11-07 18:46:04 +00:00
2019-11-13 08:38:06 +00:00
class SampleHyperOpt(IHyperOpt):
2018-11-07 18:46:04 +00:00
"""
This is a sample Hyperopt to inspire you.
2020-03-09 14:04:28 +00:00
More information in the documentation: https://www.freqtrade.io/en/latest/hyperopt/
You should:
- Rename the class name to some unique name.
- Add any methods you want to build your hyperopt.
- Add any lib you need to build your hyperopt.
2020-03-09 14:04:28 +00:00
An easier way to get a new hyperopt file is by using
`freqtrade new-hyperopt --hyperopt MyCoolHyperopt`.
You must keep:
- The prototypes for the methods: populate_indicators, indicator_space, buy_strategy_generator.
2020-03-09 14:04:28 +00:00
The methods roi_space, generate_roi_table and stoploss_space are not required
and are provided by default.
However, you may override them if you need 'roi' and 'stoploss' spaces that
differ from the defaults offered by Freqtrade.
Sample implementation of these methods will be copied to `user_data/hyperopts` when
creating the user-data directory using `freqtrade create-userdir --userdir user_data`,
or is available online under the following URL:
https://github.com/freqtrade/freqtrade/blob/develop/freqtrade/templates/sample_hyperopt_advanced.py.
2018-11-07 18:46:04 +00:00
"""
@staticmethod
def indicator_space() -> List[Dimension]:
"""
Define your Hyperopt space for searching buy strategy parameters.
"""
return [
Integer(10, 25, name='mfi-value'),
Integer(15, 45, name='fastd-value'),
Integer(20, 50, name='adx-value'),
Integer(20, 40, name='rsi-value'),
Categorical([True, False], name='mfi-enabled'),
Categorical([True, False], name='fastd-enabled'),
Categorical([True, False], name='adx-enabled'),
Categorical([True, False], name='rsi-enabled'),
Categorical(['bb_lower', 'macd_cross_signal', 'sar_reversal'], name='trigger')
]
@staticmethod
def buy_strategy_generator(params: Dict[str, Any]) -> Callable:
2018-11-07 18:46:04 +00:00
"""
Define the buy strategy parameters to be used by Hyperopt.
2018-11-07 18:46:04 +00:00
"""
def populate_buy_trend(dataframe: DataFrame, metadata: dict) -> DataFrame:
"""
Buy strategy Hyperopt will build and use.
2018-11-07 18:46:04 +00:00
"""
conditions = []
2018-11-07 18:46:04 +00:00
# GUARDS AND TRENDS
if 'mfi-enabled' in params and params['mfi-enabled']:
conditions.append(dataframe['mfi'] < params['mfi-value'])
if 'fastd-enabled' in params and params['fastd-enabled']:
conditions.append(dataframe['fastd'] < params['fastd-value'])
if 'adx-enabled' in params and params['adx-enabled']:
conditions.append(dataframe['adx'] > params['adx-value'])
if 'rsi-enabled' in params and params['rsi-enabled']:
conditions.append(dataframe['rsi'] < params['rsi-value'])
# TRIGGERS
2019-01-06 09:30:58 +00:00
if 'trigger' in params:
if params['trigger'] == 'bb_lower':
conditions.append(dataframe['close'] < dataframe['bb_lowerband'])
if params['trigger'] == 'macd_cross_signal':
conditions.append(qtpylib.crossed_above(
dataframe['macd'], dataframe['macdsignal']
))
if params['trigger'] == 'sar_reversal':
conditions.append(qtpylib.crossed_above(
dataframe['close'], dataframe['sar']
))
2018-11-07 18:46:04 +00:00
2020-03-10 15:05:33 +00:00
# Check that volume is not 0
conditions.append(dataframe['volume'] > 0)
2019-05-24 20:08:56 +00:00
if conditions:
dataframe.loc[
reduce(lambda x, y: x & y, conditions),
'buy'] = 1
2018-11-07 18:46:04 +00:00
return dataframe
return populate_buy_trend
@staticmethod
def sell_indicator_space() -> List[Dimension]:
2018-11-07 18:46:04 +00:00
"""
Define your Hyperopt space for searching sell strategy parameters.
2018-11-07 18:46:04 +00:00
"""
return [
Integer(75, 100, name='sell-mfi-value'),
Integer(50, 100, name='sell-fastd-value'),
Integer(50, 100, name='sell-adx-value'),
Integer(60, 100, name='sell-rsi-value'),
Categorical([True, False], name='sell-mfi-enabled'),
Categorical([True, False], name='sell-fastd-enabled'),
Categorical([True, False], name='sell-adx-enabled'),
Categorical([True, False], name='sell-rsi-enabled'),
Categorical(['sell-bb_upper',
'sell-macd_cross_signal',
'sell-sar_reversal'], name='sell-trigger')
2018-11-07 18:46:04 +00:00
]
2019-01-06 09:16:30 +00:00
@staticmethod
def sell_strategy_generator(params: Dict[str, Any]) -> Callable:
"""
Define the sell strategy parameters to be used by Hyperopt.
2019-01-06 09:16:30 +00:00
"""
def populate_sell_trend(dataframe: DataFrame, metadata: dict) -> DataFrame:
"""
Sell strategy Hyperopt will build and use.
2019-01-06 09:16:30 +00:00
"""
conditions = []
2019-01-06 09:16:30 +00:00
# GUARDS AND TRENDS
if 'sell-mfi-enabled' in params and params['sell-mfi-enabled']:
conditions.append(dataframe['mfi'] > params['sell-mfi-value'])
if 'sell-fastd-enabled' in params and params['sell-fastd-enabled']:
conditions.append(dataframe['fastd'] > params['sell-fastd-value'])
if 'sell-adx-enabled' in params and params['sell-adx-enabled']:
conditions.append(dataframe['adx'] < params['sell-adx-value'])
2019-01-06 09:16:30 +00:00
if 'sell-rsi-enabled' in params and params['sell-rsi-enabled']:
conditions.append(dataframe['rsi'] > params['sell-rsi-value'])
# TRIGGERS
if 'sell-trigger' in params:
if params['sell-trigger'] == 'sell-bb_upper':
conditions.append(dataframe['close'] > dataframe['bb_upperband'])
2019-01-06 09:16:30 +00:00
if params['sell-trigger'] == 'sell-macd_cross_signal':
conditions.append(qtpylib.crossed_above(
dataframe['macdsignal'], dataframe['macd']
2019-01-06 09:16:30 +00:00
))
if params['sell-trigger'] == 'sell-sar_reversal':
conditions.append(qtpylib.crossed_above(
dataframe['sar'], dataframe['close']
2019-01-06 09:16:30 +00:00
))
2020-03-10 15:05:33 +00:00
# Check that volume is not 0
conditions.append(dataframe['volume'] > 0)
2019-05-24 20:08:56 +00:00
if conditions:
dataframe.loc[
reduce(lambda x, y: x & y, conditions),
'sell'] = 1
2019-01-06 09:16:30 +00:00
return dataframe
return populate_sell_trend