Converts a Model object to a dictionary.
Return Type | Description |
---|---|
dict | A dictionary containing information from the fdl.ModelInfo() object. |
import pandas as pd
df = pd.read_csv('example_dataset.csv')
dataset_info = fdl.DatasetInfo.from_dataframe(
df=df
)
model_info = fdl.ModelInfo.from_dataset_info(
dataset_info=dataset_info,
features=[
'feature_1',
'feature_2',
'feature_3'
],
outputs=[
'output_column'
],
target='target_column',
input_type=fdl.ModelInputType.TABULAR,
model_task=fdl.ModelTask.BINARY_CLASSIFICATION
)
model_info_dict = model_info.to_dict()
{
'name': 'Example Model',
'input-type': 'structured',
'model-task': 'binary_classification',
'inputs': [
{
'column-name': 'feature_1',
'data-type': 'float'
},
{
'column-name': 'feature_2',
'data-type': 'int'
},
{
'column-name': 'feature_3',
'data-type': 'bool'
},
{
'column-name': 'target_column',
'data-type': 'int'
}
],
'outputs': [
{
'column-name': 'output_column',
'data-type': 'float'
}
],
'datasets': [],
'targets': [
{
'column-name': 'target_column',
'data-type': 'int'
}
],
'custom-explanation-names': []
}