6181253459

Committed 08 Sep 2023 06:04AM UTC coverage: 94.614% (+0.06%) from 94.556%

Build # 6181253459

Build Type

push

github-actions

Committed by

web-flow

Commit Message

Merge pull request #9159 from stash86/fix-adjust

remove old codes when we only can do partial entries

Run Details

2 of 2 new or added lines in 1 file covered. (100.0%)

19114 of 20202 relevant lines covered (94.61%)

0.95 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

94.44

/freqtrade/freqai/base_models/BasePyTorchModel.py

import logging
from abc import ABC, abstractmethod

import torch

from freqtrade.freqai.freqai_interface import IFreqaiModel
from freqtrade.freqai.torch.PyTorchDataConvertor import PyTorchDataConvertor


logger = logging.getLogger(__name__)


class BasePyTorchModel(IFreqaiModel, ABC):
    """
    Base class for PyTorch type models.
    User *must* inherit from this class and set fit() and predict() and
    data_convertor property.
    """

    def __init__(self, **kwargs):
        super().__init__(config=kwargs["config"])
        self.dd.model_type = "pytorch"
        self.device = "cuda" if torch.cuda.is_available() else "cpu"
        test_size = self.freqai_info.get('data_split_parameters', {}).get('test_size')
        self.splits = ["train", "test"] if test_size != 0 else ["train"]
        self.window_size = self.freqai_info.get("conv_width", 1)

    @property
    @abstractmethod
    def data_convertor(self) -> PyTorchDataConvertor:
        """
        a class responsible for converting `*_features` & `*_labels` pandas dataframes
        to pytorch tensors.
        """
        raise NotImplementedError("Abstract property")

1	import logging	1✔
2	from abc import ABC, abstractmethod	1✔
3
4	import torch	1✔
5
6	from freqtrade.freqai.freqai_interface import IFreqaiModel	1✔
7	from freqtrade.freqai.torch.PyTorchDataConvertor import PyTorchDataConvertor	1✔
8
9
10	logger = logging.getLogger(__name__)	1✔
11
12
13	class BasePyTorchModel(IFreqaiModel, ABC):	1✔
14	"""
15	Base class for PyTorch type models.
16	User must inherit from this class and set fit() and predict() and
17	data_convertor property.
18	"""
19
20	def __init__(self, **kwargs):	1✔
21	super().__init__(config=kwargs["config"])	1✔
22	self.dd.model_type = "pytorch"	1✔
23	self.device = "cuda" if torch.cuda.is_available() else "cpu"	1✔
24	test_size = self.freqai_info.get('data_split_parameters', {}).get('test_size')	1✔
25	self.splits = ["train", "test"] if test_size != 0 else ["train"]	1✔
26	self.window_size = self.freqai_info.get("conv_width", 1)	1✔
27
28	@property	1✔
29	@abstractmethod	1✔
30	def data_convertor(self) -> PyTorchDataConvertor:	1✔
31	"""
32	a class responsible for converting `_features` & `_labels` pandas dataframes
33	to pytorch tensors.
34	"""
35	raise NotImplementedError("Abstract property")	×

freqtrade / freqtrade / 6181253459

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous