#!/users/dsc/bin/Python3
#-*-coding: UTF-8 -*-
import numpy as np
import pandas as pd
from sklearn.datasets import load_digits#数据集 手写字
from sklearn.model_selection import train_test_split#训练集/测试机划分函数
import ml_utiliy as mlty          
digits = load_digits()
X = digits.data#特征数据
y = digits.target#标签
#训练集和测试集的划分 
x_train, x_test, y_train, y_test = train_test_split(X, y, test_size=0.2,stratify=y)
#模型载入
model_names=["MLP","LR","RFC","DTC"]
def mnTrain():
    return mlty.train("mn_test",model_names[0],x_train,y_train)
def mnTest():
    return mlty.predict_score("mn_test",model_names[0],x_test,y_test)
    
if __name__=="__main__":
    print("X.shape:{}".format(X.shape))
    print("y.shape:{}".format(y.shape))