Herramienta para la aplicación de algoritmos de aprendizaje automático a través de APIs

Abstract

En este trabajo se documenta la creación y desarrollo de una herramienta web (Python Machine Learning REST API Generator, abreviado como PyMLAPIGen). Esta herramienta tiene el objetivo de permitir a cualquier usuario no experto en el dominio de Machine Learning, entrenar algoritmos a partir de un dataset de entrada y poder aplicar dichos algoritmos entrenados a nuevos datasets. De esta forma, cualquier usuario podrá utilizar la herramienta para poder probar distintos algoritmos con diversos parámetros y así pulir y encontrar la mejor configuración para su dataset. La herramienta ofrece una interfaz de usuario a través de una aplicación web hospedada en la máquina del usuario. Además, cuenta con una API de programación REST a través de la cual se puede realizar experimentos. Un experimento genera una API REST con la que se puede interactuar a través de la herramienta web o los diversos endpoints de la API. Para generar un experimento, es necesario introducir un dataset, la categoría Machine Learning (clasificación, regresión o clustering), la columna objetivo del dataset para aprendizajes supervisados, el algoritmo y sus parámetros, los atributos para tener en cuenta del dataset y el tamaño del conjunto de entrenamiento. Una vez introducido los parámetros requeridos, un modelo será entrenado y evaluado de forma que pueda ofrecer distintas métricas, gráficas y ser utilizado para nuevas predicciones.