predictor/setup.py at master · harveybc/predictor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
from setuptools import setup, find_packages

setup(
    name='predictor',
    version='0.1.0',
    packages=find_packages(),
    entry_points={
        'console_scripts': [
            'predictor=app.main:main'
        ],
        # Plugins para el Predictor
        'predictor.plugins': [
            'default_predictor=predictor_plugins.predictor_plugin_ann:Plugin',
            'ann=predictor_plugins.predictor_plugin_ann:Plugin',
            'n_beats=predictor_plugins.predictor_plugin_n_beats:Plugin',
            'cnn=predictor_plugins.predictor_plugin_cnn:Plugin',
            'lstm=predictor_plugins.predictor_plugin_lstm:Plugin',
            'transformer=predictor_plugins.predictor_plugin_transformer:Plugin',
            'mimo=predictor_plugins.predictor_plugin_mimo:Plugin',
            'tft=predictor_plugins.predictor_plugin_tft:Plugin',
            'tcn=predictor_plugins.predictor_plugin_tcn:Plugin',
            'base=predictor_plugin.predictor_plugin_base:Plugin',
            # Binary classification plugins
            'binary_ann=predictor_plugins.binary.predictor_plugin_binary_ann:Plugin',
            'binary_cnn=predictor_plugins.binary.predictor_plugin_binary_cnn:Plugin',
            'binary_lstm=predictor_plugins.binary.predictor_plugin_binary_lstm:Plugin',
            'binary_transformer=predictor_plugins.binary.predictor_plugin_binary_transformer:Plugin',
            'binary_n_beats=predictor_plugins.binary.predictor_plugin_binary_n_beats:Plugin',
            'binary_tft=predictor_plugins.binary.predictor_plugin_binary_tft:Plugin',
            'binary_tcn=predictor_plugins.binary.predictor_plugin_binary_tcn:Plugin',
            'binary_mimo=predictor_plugins.binary.predictor_plugin_binary_mimo:Plugin',
            'binary_logistic=predictor_plugins.binary.predictor_plugin_binary_logistic:Plugin',
            # Direction classification plugins
            'direction_ann=predictor_plugins.direction.predictor_plugin_direction_ann:Plugin',
            'direction_cnn=predictor_plugins.direction.predictor_plugin_direction_cnn:Plugin',
            'direction_lstm=predictor_plugins.direction.predictor_plugin_direction_lstm:Plugin',
            'direction_transformer=predictor_plugins.direction.predictor_plugin_direction_transformer:Plugin',
            'direction_n_beats=predictor_plugins.direction.predictor_plugin_direction_n_beats:Plugin',
            'direction_tft=predictor_plugins.direction.predictor_plugin_direction_tft:Plugin',
            'direction_tcn=predictor_plugins.direction.predictor_plugin_direction_tcn:Plugin',
            'direction_mimo=predictor_plugins.direction.predictor_plugin_direction_mimo:Plugin',
            'direction_logistic=predictor_plugins.direction.predictor_plugin_direction_logistic:Plugin',
        ],
        # Plugins para la Optimización (por defecto, basado en DEAP)
        'optimizer.plugins': [
            'default_optimizer=optimizer_plugins.default_optimizer:Plugin',
            'neat_optimizer=optimizer_plugins.neat_optimizer:Plugin'
        ],
        # Plugins para el Pipeline (orquestación del flujo completo)
        'pipeline.plugins': [
            'default_pipeline=pipeline_plugins.default_pipeline:PipelinePlugin',
            'stl_pipeline=pipeline_plugins.stl_pipeline:STLPipelinePlugin',
            'binary_pipeline=pipeline_plugins.binary_pipeline:BinaryPipelinePlugin',
            'direction_pipeline=pipeline_plugins.direction_pipeline:DirectionPipelinePlugin'
        ],
        # Plugins para el Preprocesamiento (incluye process_data, ventanas deslizantes y STL)
        'preprocessor.plugins': [
            'default_preprocessor=preprocessor_plugins.default_preprocessor:PreprocessorPlugin',
            'stl_preprocessor=preprocessor_plugins.stl_preprocessor:PreprocessorPlugin'
        ],
        'target.plugins': [
            'default_target=target_plugins.default_target:TargetPlugin',
            'stl_target=target_plugins.stl_target:TargetPlugin',
            'binary_target=target_plugins.binary_target:TargetPlugin',
        ],
        # Target plugins
        'target.plugins': [
            'default_target=target_plugins.default_target:TargetPlugin',
            'stl_target=target_plugins.stl_target:TargetPlugin',
            'binary_target=target_plugins.binary_target:TargetPlugin',
            'direction_target=target_plugins.direction_target:TargetPlugin',
        ]
    },
    install_requires=[
        'build'
    ],
    author='Harvey Bastidas',
    author_email='your.email@example.com',
    description=(
        'A timeseries prediction system that supports dynamic loading of plugins for prediction, '
        'optimization, pipeline orchestration, and data pre-processing.'
    )
)