mlgen是一个帮助您轻松生成机器学习代码的工具。
mlgen的Python项目详细描述
mlgen
mlgen是一个帮助您轻松生成机器学习代码的工具。 mlgen使用“.mlm”文件格式,该格式是一个具有类似于yml语法的文件。 此工具现在支持Keras和Tensorflow2.0(不完全支持)
pip install mlgen
cli命令--->;
初始化文件mlgen -i | --init <file name>
生成特定模板(可选)mlgen -g | --gen <neural network type> --backend | -be <lib to use> -t jupyter
生成ml python文件mlgen -r .
传销文件语法--->;
文件:要创建的python文件的名称
version:使用的python版本
后端:如果要使用哪个机器学习平台
gpu:(bool)是否正在使用gpu
data:数据集的位置可以是计算机上的url/文件夹位置
split:(int)在训练和测试数据中滑动。自动转换为十进制
coloumns_功能:用于预测的列
nill_data:非分类数据类型中的基本空数据处理。可用技术删除,平均值,模式,中值
nill_data_分类:分类数据类型的基本空数据处理。可用技术删除,最大,最小
neural network_type:正在使用的神经网络类型,例如ann、cnn、lstm
layer1: number_neurons: (int) number of neurons input_dim: input dimensions of the first layer input activation: activation function being used dropout: (optional) dropout: (int) dropout percentage noise_shape: (int) noise shape (optional) seed: (int) seed value (optional) layer2: number_neurons: (int) number of neurons activation: activation function being used dropout: (optional) dropout: (int) dropout percentage noise_shape: (int) noise shape (optional) seed: (int) seed value (optional) compile: epochs: (int) number of epoch batch_size: (int) batch size verbose: (int) verbose value 0,1,2 optimizer: optimizer being used loss: loss type metrics: (array) - metrics type checkpoint: (optional) monitor: metrix type verbose: (int) batch size save_best_only: (bool) mode: mode such as min max save_model: (optional) file: file name to save model in save: save type. Available options weights and model