Fix pylint and log fmt

xgboost_ames_housing/housing.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 import argparse
-import joblib
 import logging
+import joblib
 import pandas as pd
 from sklearn.metrics import mean_absolute_error
 from sklearn.model_selection import train_test_split
@@ -55,20 +55,20 @@ def train_model(train_X,
             early_stopping_rounds=40,
             eval_set=[(test_X, test_y)])
 
-  logging.info("Best RMSE on eval: {:.2f} with {} rounds".format(
-                 model.best_score,
-                 model.best_iteration+1))
+  logging.info("Best RMSE on eval: %.2f with %d rounds",
+               model.best_score,
+               model.best_iteration+1)
   return model
 
 def eval_model(model, test_X, test_y):
   """Evaluate the model performance."""
   predictions = model.predict(test_X)
-  logging.info("mean_absolute_error={:.2f}".format(mean_absolute_error(predictions, test_y)))
+  logging.info("mean_absolute_error=%.2f", mean_absolute_error(predictions, test_y))
 
 def save_model(model, model_file):
   """Save XGBoost model for serving."""
   joblib.dump(model, model_file)
-  logging.info("Model export success {}".format(model_file))
+  logging.info("Model export success: %s", model_file)
 
 def main(args):
   (train_X, train_y), (test_X, test_y) = read_input(args.train_input)
@@ -117,5 +117,7 @@ if __name__ == '__main__':
           default=50
           )
 
+  logging.basicConfig(format='%(message)s')
+  logging.getLogger().setLevel(logging.INFO)
   main_args = parser.parse_args()
   main(main_args)