Advertisement
Guest User

Untitled

a guest
Jun 19th, 2019
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.72 KB | None | 0 0
  1. ValueError: Transformer '<class '__main__.Prepare_input'>' not supported; supported transformers are coremltools.converters.sklearn._dict_vectorizer,coremltools.converters.sklearn._one_hot_encoder,coremltools.converters.sklearn._normalizer,coremltools.converters.sklearn._standard_scaler,coremltools.converters.sklearn._imputer,coremltools.converters.sklearn._NuSVC,coremltools.converters.sklearn._NuSVR,coremltools.converters.sklearn._SVC,coremltools.converters.sklearn._SVR,coremltools.converters.sklearn._linear_regression,coremltools.converters.sklearn._LinearSVC,coremltools.converters.sklearn._LinearSVR,coremltools.converters.sklearn._logistic_regression,coremltools.converters.sklearn._random_forest_classifier,coremltools.converters.sklearn._random_forest_regressor,coremltools.converters.sklearn._decision_tree_classifier,coremltools.converters.sklearn._decision_tree_regressor,coremltools.converters.sklearn._gradient_boosting_classifier,coremltools.converters.sklearn._gradient_boosting_regressor.
  2.  
  3. class Prepare_input(BaseEstimator):
  4. def __init__(self, *args, **kwargs):
  5. return super().__init__(*args, **kwargs)
  6.  
  7. def fit(self, x, y=None):
  8. return self
  9.  
  10. def transform(self, X, y=None):
  11. x= self.extract_features(x)
  12. return x
  13.  
  14. def extract_features(signal):
  15. print("transforming...")
  16. signal = some_modif(signal)
  17. return signal
  18.  
  19. classifier1= RandomForestClassifier(max_depth=700, n_estimators=100, random_state = 42)
  20. pipeline1 = Pipeline([('Prepare_input', input_transform), ('classifier', classifier1)])
  21.  
  22. pipeline1.fit(train_data, train_labels)
  23.  
  24. coreml_model = coremltools.converters.sklearn.convert(pipeline1)
  25. coreml_model.save('PPG_classif.mlmodel')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement