Add sklearn to coreml-converter

tools/coreml-converter

@@ -21,6 +21,7 @@ fi
 export PYTHONUSERBASE=${third_party}/${identifier}/build/pypi
 export PATH=$PATH:${PYTHONUSERBASE}/bin
 pip install --user onnx
+pip install --user sklearn
 pip install --user ${third_party}/${identifier}
 
 python ${tools}/coreml-converter.py $@

tools/coreml-converter.py

@@ -10,3 +10,9 @@ if extension == '.h5':
     import coremltools
     coreml_model = coremltools.converters.keras.convert(file)
     coreml_model.save(base + '.mlmodel')
+elif extension == '.pkl':
+    import coremltools
+    import sklearn
+    sklearn_model = sklearn.externals.joblib.load(file)
+    coreml_model = coremltools.converters.sklearn.convert(sklearn_model)
+    coreml_model.save(base + '.mlmodel')