change the feature transform function
using ezplot to plot non-linear function
add comment for this scrip