def pipeline(name): if name == 'naive_bayes_counts': return pipelines.naive_bayes_counts() elif name == 'naive_bayes_tfidf': return pipelines.naive_bayes_tfidf() elif name == 'naive_bayes_tfidf_stopwords': return pipelines.naive_bayes_tfidf_stopwords() elif name == 'naive_bayes_bigram': return pipelines.naive_bayes_bigram() elif name == 'naive_bayes_trigram': return pipelines.naive_bayes_trigram() elif name == 'naive_bayes_counts_lex': return pipeline_with_lexicon.naive_bayes_counts_lex() elif name == 'svm_libsvc_counts': return pipelines.svm_libsvc_counts() elif name == 'svm_libsvc_tfidf': return pipelines.svm_libsvc_tfidf() elif name == 'svm_libsvc_tfidf_stopwords': return pipelines.svm_libsvc_tfidf_stopwords() elif name == 'svm_libsvc_counts_bigram': return pipelines.svm_libsvc_counts_bigram() elif name == 'svm_libsvc_embed': return pipelines.svm_libsvc_embed() elif name == 'svm_sigmoid_embed': return pipelines.svm_sigmoid_embed() elif name == 'random_forest_tfidf': return pipelines.random_forest_tfidf() elif name == 'random_forest_tfidf_stopwords': return pipelines.random_forest_tfidf_stopwords() elif name == 'random_forest_embed': return pipelines.random_forest_embed() else: raise ValueError( "pipeline name is unknown. You can add a custom pipeline in 'pipelines'" )
def pipeline(name): if name == 'naive_bayes': return pipelines.naive_bayes() elif name == 'svm_libsvc_counts': return pipelines.svm_libsvc_counts() elif name == 'svm_libsvc_tfidf': return pipelines.svm_libsvc_tfidf() elif name == 'svm_libsvc_embed': return pipelines.svm_libsvc_embed() elif name == 'svm_sigmoid_embed': return pipelines.svm_sigmoid_embed() else: raise ValueError("pipeline name is unknown. You can add a custom pipeline in 'pipelines'")