如何使用mysql数据库作为机器学习的数据集

Question

我已经完成了一个项目，使用 django 和 mysql 数据库来收集销售数据，我想从数据中进行一些预测，例如使用 ANN/SVM 进行需求预测

是否可以直接将 mysql 数据库作为输入，或者我应该将它们转换为 csv

Answer 1

如果您的服务器/机器有足够的 RAM，您可以直接从 MySQL 读取数据到 Pandas DataFrame 中，然后将该 DataFrame 提供给 sklearn/Keras/Tensorflow。

演示：

from sqlalchemy import create_engine
import pymysql
import pandas as pd

db_connection = 'mysql+pymysql://mysql_user:mysql_password@mysql_host/mysql_db'
conn = create_engine(db_connection)

df = pd.read_sql("select * from tab_name", conn)