Add Parquet framework and model type

This commit is contained in:
allegroai 2020-05-13 20:32:49 +03:00
parent 9130a06210
commit 0dec7c0e9e

View File

@ -47,6 +47,7 @@ class Framework(Options):
paddlepaddle = 'PaddlePaddle'
scikitlearn = 'ScikitLearn'
xgboost = 'XGBoost'
parquet = 'Parquet'
__file_extensions_mapping = {
'.pb': (tensorflow, tensorflowjs, onnx, ),
@ -74,6 +75,7 @@ class Framework(Options):
'.cfg': (darknet, ),
'__model__': (paddlepaddle, ),
'.pkl': (scikitlearn, keras, xgboost),
'.parquet': (parquet),
}
@classmethod