Как в создании модели добавить преобразование строки в число типа float?

Я новичок в питоне(за код не судите строго в основном до этого писал на с++ (в основном олимпиады)), и сейчас решил попробовать написать простую (точнее тупую) нейронку на питоне. Использую tensorflow и keras. На вход нейронке должен подаваться массив из 9 чисел и она должна распределять массив на 2 группы. Вот так я создаю массив:

   with open('yer_inf.txt', 'r') as f:
    c = []
    for line in f:
        c.append(float(line))
        if (len(c) == 9):
            b.append(c)
            # c.clear()
            c = []
    f.close()
data = np.array(b) #data.shape={664,9}

Вот так я задаю базовую модель:

model = tf.keras.Sequential([
    layers.Dense(18, activation='relu', input_dim=9),
    layers.Dense(18, activation='relu'),
    layers.Dense(1, activation='softmax')
]) 

Ну и у меня выдаётся ошибку при попытке model.fit(): Unimplemented: Cast string to float is not supported Traceback (most recent call last). Но как задать преобразование я не могу найти в документации tensorflow.

Помогите, пожалуйста.


Ответы (0 шт):