Градиентный спуск с возвратом (поиск строки) с функцией потерь MSE в python

Question

Я пытаюсь осуществить возврат с градиентным спуском, у меня работает часть с градиентным спуском, и я не знаю, как go откатить назад.

шаги для возврата

код структурирован следующим образом:

def backtracking (X, theta, alpha=0.3, beta=0.8, t =1):

    while ???:
        t *= beta
    return t 

Может ли кто-нибудь помочь с заполнением условия поиска строки? Я понимаю математику, но в настоящее время возникают проблемы с ее реализацией в python.

Мой псевдокод с градиентным спуском подобен следующему:

def gradient_descent(X, y, theta, learningRate, max_iter):
    for i in max_iter:
       theta = theta - learningRate*(XT.dot(np.dot (X, theta) - y)
       MSELoss =  (1/(2*m)) * np.sum(np.square(prediction-y))

Пожалуйста, помогите и большое спасибо !!!