File size: 414 Bytes
e8ac98a |
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 |
import numpy as np
from setuptools import setup
from Cython.Build import cythonize
setup(
ext_modules = cythonize([
# Model Builder
'model_builder.pyx',
# Data Processor
'data_processor.pyx',
# Post Processor
'post_processor.pyx',
# Trainer
'trainer.pyx',
# Main
'main.pyx'
]),
include_dirs = [ np.get_include() ]
)
|