Как обучить мою нейронку?

Помогите мне с обучением нейронной сети (код ниже).

Нейронка должна распознавать мессенджеры (ещё мессенджеры добавлю позже).

Далее, я уже буду взаимодействовать с тем, что распознала нейронка.

Я прошу просто показать, как обучается нейронная сеть и что делать после её обучения.

Код :

import numpy as np

INPUT_DIM = 1

OUT_DIM = 3

H_DIM = 5

x = np.random.randn(INPUT_DIM)

W1 = np.random.randn(INPUT_DIM,H_DIM)

b1 = np.random.randn(H_DIM)

W2 = np.random.randn(H_DIM,OUT_DIM)

b2 = np.random.randn(OUT_DIM)

def relu(t):

    return np.maximum(t,0)

def softmax(t):

    out = np.exp(t)

    return out / np.sum(out)

def predict(x):

    t1 = x @ W1 + b1

    h1 = relu(t1)

    t2 = h1 @ W2 + b2

    z = softmax(t2)

    return z

probs = predict(x=x)

pred_class = np.argmax(probs)

class_names = ['WhatsApp','Telegramm','Vk']

print('Messanger : ',class_names[pred_class])