Updated logisitic regression yaml