Используя Людвига в Colab, пытаемся выяснить, как создать встроенный файл model_definition.yaml для использования при построении модели. - PullRequest
Новая
       22

Используя Людвига в Colab, пытаемся выяснить, как создать встроенный файл model_definition.yaml для использования при построении модели.

0 голосов
/ 15 марта 2019

Таким образом, документация предусматривает, что файл model_definition.yaml необходим для обучения и тестирования моделей в Людвиге от Uber.Я пытаюсь использовать его в настраиваемом наборе данных, в котором у меня есть файл CSV, который также был прочитан в pandas в качестве блока данных.Я пытался создать определение модели_1001 *

Я пытался использовать этот подход из ответов github от людей Людвига: 

import yaml

model_definition = {}
model_definition['input_features'] = [{'name': acoustic_data, type: category}, {'name': time_to_failure, type: numerical}]
model_definition['output_features'] = [{'name': time_to_failure.train.csv, 'type': numerical,loss:mean_absolute_error}]
model_definition['training'] = {}
ludwig_model = LudwigModel(model_definition)
train_stats = ludwig_model.train(data_df=LANL_data)

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

Ошибка - синтаксическая ошибка.

1 Ответ

0 голосов
/ 16 апреля 2019

Это не дало мне синтаксической ошибки 

dict = {'input_features': [{'name': 'Open_Price', 'type': 'numerical'}], 'output_features': [{'name': 'Closing_Price', 'type': 'numerical'}]}
model_definition = dict 
ludwig_model = LudwigModel(model_definition)
...