最简单的机器学习入门实践——鸢尾花识别

定义一个函数

def MyIrisType():
    import numpy as np
    import tensorflow as tf
    import pandas as pd
    from sklearn.datasets import load_iris
    import mglearn

    #加载数据集
    iris_dataset = load_iris()
    from sklearn.model_selection import train_test_split

       """

                   train_test_split(),随机划分样本数据、训练数据。    

                  该函数一般有5个参数，1）待划分样本数据，2）待划分样本数据结果，3）测试数据占样本数据比例test_size,如果为整数，则为大小，小数则为占比，4）设置                    随机数种子random_state，若为0则不填，5）stratify，保持split前类的分布，划分前后一致。