config_NVT_ens.txt

name: "0_transformworkflow"
input {
  name: "item_id"
  data_type: TYPE_INT32
  dims: -1
  dims: 1
}
input {
  name: "category"
  data_type: TYPE_INT32
  dims: -1
  dims: 1
}
input {
  name: "day"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
input {
  name: "age_days"
  data_type: TYPE_FP32
  dims: -1
  dims: 1
}
input {
  name: "weekday_sin"
  data_type: TYPE_FP32
  dims: -1
  dims: 1
}
input {
  name: "session_id"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "session_id"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "day-first"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "item_id-count"
  data_type: TYPE_INT32
  dims: -1
  dims: 1
}
output {
  name: "item_id-list__values"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "item_id-list__lengths"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "category-list__values"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "category-list__lengths"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "age_days-list__values"
  data_type: TYPE_FP32
  dims: -1
  dims: 1
}
output {
  name: "age_days-list__lengths"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "weekday_sin-list__values"
  data_type: TYPE_FP32
  dims: -1
  dims: 1
}
output {
  name: "weekday_sin-list__lengths"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
parameters {
  key: "cats"
  value {
  }
}
parameters {
  key: "conts"
  value {
  }
}
parameters {
  key: "output_model"
  value {
  }
}
parameters {
  key: "python_module"
  value {
    string_value: "merlin.systems.triton.models.workflow_model"
  }
}
backend: "python"


### CONGIG OF ensemble_model

name: "ensemble_model"
platform: "ensemble"
input {
  name: "item_id"
  data_type: TYPE_INT32
  dims: -1
  dims: 1
}
input {
  name: "category"
  data_type: TYPE_INT32
  dims: -1
  dims: 1
}
input {
  name: "day"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
input {
  name: "age_days"
  data_type: TYPE_FP32
  dims: -1
  dims: 1
}
input {
  name: "weekday_sin"
  data_type: TYPE_FP32
  dims: -1
  dims: 1
}
input {
  name: "session_id"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "session_id"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "day-first"
  data_type: TYPE_INT64
  dims: -1
  dims: 1
}
output {
  name: "item_id-count"
  data_type: TYPE_INT32
  dims: -1
  dims: 1
}
output {
  name: "item_id-list__values"
  data_type: TYPE_INT64
  dims: -1
  dims: -1
}
output {
  name: "item_id-list__lengths"
  data_type: TYPE_INT32
  dims: -1
  dims: -1
}
output {
  name: "category-list__values"
  data_type: TYPE_INT64
  dims: -1
  dims: -1
}
output {
  name: "category-list__lengths"
  data_type: TYPE_INT32
  dims: -1
  dims: -1
}
output {
  name: "age_days-list__values"
  data_type: TYPE_FP32
  dims: -1
  dims: -1
}
output {
  name: "age_days-list__lengths"
  data_type: TYPE_INT32
  dims: -1
  dims: -1
}
output {
  name: "weekday_sin-list__values"
  data_type: TYPE_FP32
  dims: -1
  dims: -1
}
output {
  name: "weekday_sin-list__lengths"
  data_type: TYPE_INT32
  dims: -1
  dims: -1
}
ensemble_scheduling {
  step {
    model_name: "0_transformworkflow"
    model_version: -1
    input_map {
      key: "age_days"
      value: "age_days"
    }
    input_map {
      key: "category"
      value: "category"
    }
    input_map {
      key: "day"
      value: "day"
    }
    input_map {
      key: "item_id"
      value: "item_id"
    }
    input_map {
      key: "session_id"
      value: "session_id"
    }
    input_map {
      key: "weekday_sin"
      value: "weekday_sin"
    }
    output_map {
      key: "age_days-list__lengths"
      value: "age_days-list__lengths"
    }
    output_map {
      key: "age_days-list__values"
      value: "age_days-list__values"
    }
    output_map {
      key: "category-list__lengths"
      value: "category-list__lengths"
    }
    output_map {
      key: "category-list__values"
      value: "category-list__values"
    }
    output_map {
      key: "day-first"
      value: "day-first"
    }
    output_map {
      key: "item_id-count"
      value: "item_id-count"
    }
    output_map {
      key: "item_id-list__lengths"
      value: "item_id-list__lengths"
    }
    output_map {
      key: "item_id-list__values"
      value: "item_id-list__values"
    }
    output_map {
      key: "session_id"
      value: "session_id"
    }
    output_map {
      key: "weekday_sin-list__lengths"
      value: "weekday_sin-list__lengths"
    }
    output_map {
      key: "weekday_sin-list__values"
      value: "weekday_sin-list__values"
    }
  }
}