tebakaja's picture
[ update ]: Cythonize Machine Learning Utilities
e8ac98a
raw
history blame
414 Bytes
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() ]
)