Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: ML Model Backend Implementation #1896

Merged
merged 33 commits into from
Feb 17, 2021
Merged
Show file tree
Hide file tree
Changes from 27 commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
7b724a5
Merge pull request #1 from linkedin/master
RyanHolstien Apr 9, 2020
8475ad3
Merge pull request #2 from linkedin/master
RyanHolstien Jul 2, 2020
c18c55b
Merge pull request #3 from linkedin/master
RyanHolstien Sep 16, 2020
dc474a7
Merge pull request #4 from linkedin/master
RyanHolstien Sep 25, 2020
c7d519c
feat(1877): add backend implementation for ML Model
Sep 25, 2020
d347b96
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Sep 25, 2020
81a342e
Merge pull request #5 from linkedin/master
RyanHolstien Sep 25, 2020
02a9970
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Sep 25, 2020
26a2908
update mce cli examples
Sep 25, 2020
8c897fb
changes for review comments
Oct 6, 2020
87ab115
modify owners field in index per review comments
Oct 7, 2020
938b8d2
Merge pull request #6 from linkedin/master
RyanHolstien Oct 7, 2020
42dc2f5
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Oct 7, 2020
10f5d89
update package of GMS factories and use BaseSearchableClient
Oct 7, 2020
29eab26
fix for checkstyle
Oct 7, 2020
391304c
remove unused comma analyzers and tokenizers
Oct 8, 2020
3ad1b2f
Merge pull request #7 from linkedin/master
RyanHolstien Oct 8, 2020
912b2a6
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Oct 8, 2020
69bbcbd
fix endpoint naming for evaluation data
Oct 9, 2020
059def7
fixes for IndexBuilder per review comments
Oct 14, 2020
2bc1963
update IndexBuilder with efficient setting of urn derived fields
Oct 14, 2020
075ecfa
remove unnecessary filters and fields from index builder and relation…
Jan 25, 2021
3f83e7f
Merge pull request #8 from linkedin/master
RyanHolstien Jan 25, 2021
21b3c23
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Jan 26, 2021
5646b85
remove unused imports, fix formatting, and add new integration test t…
Jan 26, 2021
8215a60
add custom analyzer to split up URN components and fix typos
Jan 27, 2021
25daf59
update query template to include urn components
Jan 27, 2021
1735ff3
consolidate query template to one operator
Feb 2, 2021
ea9434c
Merge pull request #9 from linkedin/master
RyanHolstien Feb 9, 2021
86b54cb
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Feb 9, 2021
f4e27af
update sanity test to master
Feb 10, 2021
22c233b
Merge pull request #10 from linkedin/master
RyanHolstien Feb 16, 2021
c2dbf80
Merge branch 'master' into feature/DATAHUB-1877-MLModelBackendImpl
Feb 16, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docker/elasticsearch-setup/create-indices.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,3 +16,4 @@ create_index corpuserinfodocument corp-user/settings.json corp-user/mappings.jso
create_index dashboarddocument dashboard/settings.json dashboard/mappings.json
create_index dataprocessdocument data-process/settings.json data-process/mappings.json
create_index datasetdocument dataset/settings.json dataset/mappings.json
create_index mlmodeldocument ml-model/settings.json ml-model/mappings.json
391 changes: 391 additions & 0 deletions gms/api/src/main/idl/com.linkedin.ml.mlModels.restspec.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,391 @@
{
"name" : "mlModels",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels",
"schema" : "com.linkedin.ml.MLModel",
"doc" : "generated from: com.linkedin.metadata.resources.ml.MLModels",
"collection" : {
"identifier" : {
"name" : "model",
"type" : "com.linkedin.ml.MLModelKey",
"params" : "com.linkedin.restli.common.EmptyRecord"
},
"supports" : [ "batch_get", "get", "get_all" ],
"methods" : [ {
"method" : "get",
"parameters" : [ {
"name" : "aspects",
"type" : "{ \"type\" : \"array\", \"items\" : \"string\" }",
"default" : "[]"
} ]
}, {
"method" : "batch_get",
"parameters" : [ {
"name" : "aspects",
"type" : "{ \"type\" : \"array\", \"items\" : \"string\" }",
"default" : "[]"
} ]
}, {
"method" : "get_all",
"parameters" : [ {
"name" : "aspects",
"type" : "{ \"type\" : \"array\", \"items\" : \"string\" }",
"default" : "[]"
}, {
"name" : "filter",
"type" : "com.linkedin.metadata.query.Filter",
"optional" : true
}, {
"name" : "sort",
"type" : "com.linkedin.metadata.query.SortCriterion",
"optional" : true
} ],
"pagingSupported" : true
} ],
"finders" : [ {
"name" : "search",
"parameters" : [ {
"name" : "input",
"type" : "string"
}, {
"name" : "aspects",
"type" : "{ \"type\" : \"array\", \"items\" : \"string\" }",
"default" : "[]"
}, {
"name" : "filter",
"type" : "com.linkedin.metadata.query.Filter",
"optional" : true
}, {
"name" : "sort",
"type" : "com.linkedin.metadata.query.SortCriterion",
"optional" : true
} ],
"metadata" : {
"type" : "com.linkedin.metadata.query.SearchResultMetadata"
},
"pagingSupported" : true
} ],
"actions" : [ {
"name" : "autocomplete",
"parameters" : [ {
"name" : "query",
"type" : "string"
}, {
"name" : "field",
"type" : "string"
}, {
"name" : "filter",
"type" : "com.linkedin.metadata.query.Filter"
}, {
"name" : "limit",
"type" : "int"
} ],
"returns" : "com.linkedin.metadata.query.AutoCompleteResult"
}, {
"name" : "backfill",
"parameters" : [ {
"name" : "urn",
"type" : "string"
}, {
"name" : "aspects",
"type" : "{ \"type\" : \"array\", \"items\" : \"string\" }",
"default" : "[]"
} ],
"returns" : "com.linkedin.metadata.restli.BackfillResult"
}, {
"name" : "getSnapshot",
"parameters" : [ {
"name" : "urn",
"type" : "string"
}, {
"name" : "aspects",
"type" : "{ \"type\" : \"array\", \"items\" : \"string\" }",
"default" : "[]"
} ],
"returns" : "com.linkedin.metadata.snapshot.MLModelSnapshot"
}, {
"name" : "ingest",
"parameters" : [ {
"name" : "snapshot",
"type" : "com.linkedin.metadata.snapshot.MLModelSnapshot"
} ]
} ],
"entity" : {
"path" : "/mlModels/{model}",
"subresources" : [ {
"name" : "caveatsAndRecommendations",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/caveatsAndRecommendations",
"schema" : "com.linkedin.ml.metadata.CaveatsAndRecommendations",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/caveatsAndRecommendations\n\ngenerated from: com.linkedin.metadata.resources.ml.CaveatsAndRecommendationsResource",
"collection" : {
"identifier" : {
"name" : "caveatsAndRecommendationsId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/caveatsAndRecommendations/{caveatsAndRecommendationsId}"
}
}
}, {
"name" : "cost",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/cost",
"schema" : "com.linkedin.common.Cost",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/cost\n\ngenerated from: com.linkedin.metadata.resources.ml.CostResource",
"collection" : {
"identifier" : {
"name" : "costId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/cost/{costId}"
}
}
}, {
"name" : "deprecation",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/deprecation",
"schema" : "com.linkedin.common.Deprecation",
"doc" : "Rest.li entry point: /mlmodels/{mlModelKey}/deprecation\n\ngenerated from: com.linkedin.metadata.resources.ml.DeprecationResource",
"collection" : {
"identifier" : {
"name" : "deprecationId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/deprecation/{deprecationId}"
}
}
}, {
"name" : "ethicalConsiderations",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/ethicalConsiderations",
"schema" : "com.linkedin.ml.metadata.EthicalConsiderations",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/ethicalConsiderations\n\ngenerated from: com.linkedin.metadata.resources.ml.EthicalConsiderationsResource",
"collection" : {
"identifier" : {
"name" : "ethicalConsiderationsId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/ethicalConsiderations/{ethicalConsiderationsId}"
}
}
}, {
"name" : "evaluationData",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/evaluationData",
"schema" : "com.linkedin.ml.metadata.EvaluationData",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/evaluationData\n\ngenerated from: com.linkedin.metadata.resources.ml.EvaluationDataResource",
"collection" : {
"identifier" : {
"name" : "evaluationDataId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/evaluationData/{evaluationDataId}"
}
}
}, {
"name" : "institutionalMemory",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/institutionalMemory",
"schema" : "com.linkedin.common.InstitutionalMemory",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/institutionalMemory\n\ngenerated from: com.linkedin.metadata.resources.ml.InstitutionalMemoryResource",
"collection" : {
"identifier" : {
"name" : "institutionalMemoryId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/institutionalMemory/{institutionalMemoryId}"
}
}
}, {
"name" : "intendedUse",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/intendedUse",
"schema" : "com.linkedin.ml.metadata.IntendedUse",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/intendedUse\n\ngenerated from: com.linkedin.metadata.resources.ml.IntendedUseResource",
"collection" : {
"identifier" : {
"name" : "intendedUseId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/intendedUse/{intendedUseId}"
}
}
}, {
"name" : "metrics",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/metrics",
"schema" : "com.linkedin.ml.metadata.Metrics",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/metrics\n\ngenerated from: com.linkedin.metadata.resources.ml.MetricsResource",
"collection" : {
"identifier" : {
"name" : "metricsId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/metrics/{metricsId}"
}
}
}, {
"name" : "mlModelFactorPrompts",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/mlModelFactorPrompts",
"schema" : "com.linkedin.ml.metadata.MLModelFactorPrompts",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/mlModelFactorPrompts\n\ngenerated from: com.linkedin.metadata.resources.ml.MLModelFactorPromptsResource",
"collection" : {
"identifier" : {
"name" : "mlModelFactorPromptsId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/mlModelFactorPrompts/{mlModelFactorPromptsId}"
}
}
}, {
"name" : "ownership",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/ownership",
"schema" : "com.linkedin.common.Ownership",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/ownership\n\ngenerated from: com.linkedin.metadata.resources.ml.OwnershipResource",
"collection" : {
"identifier" : {
"name" : "ownershipId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/ownership/{ownershipId}"
}
}
}, {
"name" : "quantitativeAnalyses",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/quantitativeAnalyses",
"schema" : "com.linkedin.ml.metadata.QuantitativeAnalyses",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/quantitativeAnalyses\n\ngenerated from: com.linkedin.metadata.resources.ml.QuantitativeAnalysesResource",
"collection" : {
"identifier" : {
"name" : "quantitativeAnalysesId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/quantitativeAnalyses/{quantitativeAnalysesId}"
}
}
}, {
"name" : "sourceCode",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/sourceCode",
"schema" : "com.linkedin.ml.metadata.SourceCode",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/sourceCode\n\ngenerated from: com.linkedin.metadata.resources.ml.SourceCodeResource",
"collection" : {
"identifier" : {
"name" : "sourceCodeId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/sourceCode/{sourceCodeId}"
}
}
}, {
"name" : "trainingData",
"namespace" : "com.linkedin.ml",
"path" : "/mlModels/{model}/trainingData",
"schema" : "com.linkedin.ml.metadata.TrainingData",
"doc" : "Rest.li entry point: /mlModels/{mlModelKey}/trainingData\n\ngenerated from: com.linkedin.metadata.resources.ml.TrainingDataResource",
"collection" : {
"identifier" : {
"name" : "trainingDataId",
"type" : "long"
},
"supports" : [ "create", "get" ],
"methods" : [ {
"method" : "create"
}, {
"method" : "get"
} ],
"entity" : {
"path" : "/mlModels/{model}/trainingData/{trainingDataId}"
}
}
} ]
}
}
}
Loading