太骚了，用 Excel 玩机器学习

你们好，最近发现了一个好玩的Python库，它可以将训练好的机器学习模型转换为Java、C、JavaScript、Go、Ruby，VBA 本地代码，可以让连Python和机器学习一无所知的同学也能感受预测的神奇。

推荐文章

• 李宏毅《机器学习》国语课程(2022)来了

• 有人把吴恩达老师的机器学习和深度学习做成了中文版

• 上瘾了，最近又给公司撸了一个可视化大屏(附源码)

• 如此优雅，4款 Python 自动数据分析神器真香啊

• 梳理半月有余，精心准备了17张知识思维导图，这次要讲清统计学

• 年终汇总：20份可视化大屏模板，直接套用真香(文末附源码)

先看效果哈

输入特征数据，点击开始预测，即可输出类别概率

这个小东西，感觉还能玩出别的花样，但不清楚到底有什么场景。写这篇文章的时候，我对Excel的 VBA 一无所知，百度了几个用法就开搞了。现在我也能用 VBA 写一些函数和过程了，这也许是最大的收货吧。

m2cgen

m2cgen（Model 2 Code Generator）是一个轻量级代码生成器，它可以将训练好的机器学习模型转换成无需依赖库的本地代码。

m2cgen目前支持的模型还蛮多的，常用常见的都包括了：

使用方法

m2cgen的安装非常方便，直接pip：

pip install m2cgen

使用，先用XGBClassifier训练一个模型

# import packages
import pandas as pd
import numpy as np
import os re
from random import sample
from sklearn import datasets
from xgboost import XGBClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
import pickle
import m2cgen as m2c

seed = 2020
test_size = 0.3
X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=test_size, random_state=seed)
# fit model on training data
model = XGBClassifier()
model.fit(X_train, y_train)

将模型转为VBA

code = m2c.export_to_visual_basic(model, function_name = 'pred')

VBA的英文全称是Visual Basic for Applications，是一门标准的宏语言，通常使用来实现Excel中没有提供的功能、编写自定义函数、实现自动化功能等。VBA语言不能单独运行，只能被office软件（如：Word、Excel等）所调用。

生成的VBA长这样：

实话实话，我对VBA还是小白，也懒得深究，于是就把代码改成了过程，这里就不贴了，我把改好之后的代码放到了github。有VBA大神可以帮忙，欢迎提交 PR。

https://github.com/tjxj/excel2ml

想省事的同学直接去copy即可

Excel

点击excel菜单中的【开发工具】（如果没有开发工具选项卡，请参考：excel开发工具选项卡在哪？原来需要自己添加），然后在代码选项卡中点击【Visual Basic】 即可，也可以直接Alt + F11 唤出：

点击插入-模块，把改好的 VBA 代码贴进去即可

保存后退出，然后回到sheet页，开发工具-插入-按钮编辑好文字后，右键指定宏，选中我们刚才保存的那个即可。

然后就一切OK了。

技术交流

欢迎转载、收藏、有所收获点赞支持一下！

目前开通了技术交流群，群友已超过2000人，添加时最好的备注方式为：来源+兴趣方向，方便找到志同道合的朋友

• 方式①、发送如下图片至微信，长按识别，后台回复：加群；
• 方式②、添加微信号：dkl88191，备注：来自CSDN
• 方式③、微信搜索公众号：Python学习与数据挖掘，后台回复：加群