mirror of
https://github.com/clearml/clearml
synced 2025-02-07 21:33:25 +00:00
Add Parquet framework and model type
This commit is contained in:
parent
9130a06210
commit
0dec7c0e9e
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user