Changing log file extension

Fixing print statement and add a separator

Calculating class weights

Fixing a parameter

Adding training section to README.md

Defining and implementing build unet model function

Optimizing import statements

Fixing README.md

Fixing lift check issue - 01

Fixing a minor issue