ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 15:45:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Parquet framework and model type

Browse Source
This commit is contained in:
allegroai 2020-05-13 20:32:49 +03:00
parent 9130a06210
commit 0dec7c0e9e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trains/model.py
View File

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