pandas - 在 Colab 中使用 Ludwig 试图弄清楚如何创建内联的 model_definition.yaml 文件以用于构建模型

Question

因此，文档提供了一个 model_definition.yaml 文件来训练和测试 Uber 的 ludwig 模型。我正在尝试在自定义数据集上使用它，我有一个 CSV 文件，该文件也作为数据框读入 pandas。我试图创建一个model_definition

我一直在尝试使用路德维希人的 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)

有关如何创建 model_definitions 的文档需要更详细。

错误是语法错误。

Answer 1

这没有给我语法错误

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

Answer 2

你最好引用字典中的值。