Coverage for credoai/artifacts/model/constants_model.py: 100%

1from credoai.utils.model_utils import ( 

2 validate_sklearn_like, 

3 validate_keras_clf, 

4 validate_dummy, 

5) 

6 

7SKLEARN_LIKE_FRAMEWORKS = ["sklearn", "xgboost"] 

8MODEL_TYPES = [ 

9 "REGRESSION", 

10 "CLASSIFICATION", 

11 "BINARY_CLASSIFICATION", 

12 "MULTICLASS_CLASSIFICATION", 

13 "COMPARISON", 

14] 

15 

16# Multilayer Perceptron 

17MLP_FRAMEWORKS = ["keras"] 

18 

19FRAMEWORK_VALIDATION_FUNCTIONS = { 

20 "sklearn": validate_sklearn_like, 

21 "xgboost": validate_sklearn_like, 

22 "keras": validate_keras_clf, 

23 "credoai": validate_dummy, 

24 # check on tensorflow generic, given validation strictness 

25}