<s> COLOUR_FIGURE = False <EOL> from matplotlib import pyplot as plt <EOL> from sklearn . datasets import load_iris <EOL> data = load_iris ( ) <EOL> features = data . data <EOL> feature_names = data . feature_names <EOL> target = data . target <EOL> target_names = data . target_names <EOL> labels = target_names [ target ] <EOL> is_setosa = ( labels == '<STR_LIT>' ) <EOL> features = features [ ~ is_setosa ] <EOL> labels = labels [ ~ is_setosa ] <EOL> is_virginica = ( labels == '<STR_LIT>' ) <EOL> t = <NUM_LIT> <EOL> t2 = <NUM_LIT> <EOL> f0 , f1 = <NUM_LIT:3> , <NUM_LIT:2> <EOL> if COLOUR_FIGURE : <EOL> area1c = ( <NUM_LIT:1.> , <NUM_LIT> , <NUM_LIT> ) <EOL> area2c = ( <NUM_LIT> , <NUM_LIT> , <NUM_LIT:1.> ) <EOL> else : <EOL> area1c = ( <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:1> ) <EOL> area2c = ( <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ) <EOL> x0 = features [ : , f0 ] . min ( ) * <NUM_LIT> <EOL> x1 = features [ : , f0 ] . max ( ) * <NUM_LIT> <EOL> y0 = features [ : , f1 ] . min ( ) * <NUM_LIT> <EOL> y1 = features [ : , f1 ] . max ( ) * <NUM_LIT> <EOL> fig , ax = plt . subplots ( ) <EOL> ax . fill_between ( [ t , x1 ] , [ y0 , y0 ] , [ y1 , y1 ] , color = area2c ) <EOL> ax . fill_between ( [ x0 , t ] , [ y0 , y0 ] , [ y1 , y1 ] , color = area1c ) <EOL> ax . plot ( [ t , t ] , [ y0 , y1 ] , '<STR_LIT>' , lw = <NUM_LIT:2> ) <EOL> ax . plot ( [ t2 , t2 ] , [ y0 , y1 ] , '<STR_LIT>' , lw = <NUM_LIT:2> ) <EOL> ax . scatter ( features [ is_virginica , f0 ] , <EOL> features [ is_virginica , f1 ] , c = '<STR_LIT:b>' , marker = '<STR_LIT:o>' , s = <NUM_LIT> ) <EOL> ax . scatter ( features [ ~ is_virginica , f0 ] , <EOL> features [ ~ is_virginica , f1 ] , c = '<STR_LIT:r>' , marker = '<STR_LIT:x>' , s = <NUM_LIT> ) <EOL> ax . set_ylim ( y0 , y1 ) <EOL> ax . set_xlim ( x0 , x1 ) <EOL> ax . set_xlabel ( feature_names [ f0 ] ) <EOL> ax . set_ylabel ( feature_names [ f1 ] ) <EOL> fig . tight_layout ( ) <EOL> fig . savefig ( '<STR_LIT>' ) </s>
