Skip to content

Commit

Permalink
2023-10-25-bert_ft_qqp_79_jeevesh8_en (#14040)
Browse files Browse the repository at this point in the history
* Add model 2023-10-25-bert_ft_qqp_68_jeevesh8_en

* Add model 2023-10-25-bert_base_uncased_sst_2_32_13_30_en

* Add model 2023-10-25-boss_sentiment_bert_base_uncased_en

* Add model 2023-10-25-bert_mini_finetuned_sst2_en

* Add model 2023-10-25-bert_ft_qqp_95_jeevesh8_en

* Add model 2023-10-25-bert_tiny_finetuned_mnli_en

* Add model 2023-10-25-bert_base_dutch_cased_finetuned_dt_en

* Add model 2023-10-25-bert_tiny_finetuned_mrpc_en

* Add model 2023-10-25-goog_bert_ft_cola_46_en

* Add model 2023-10-25-bert_tiny_finetuned_qnli_en

* Add model 2023-10-25-dk_emotion_bert_class_en

* Add model 2023-10-25-goog_bert_ft_cola_47_en

* Add model 2023-10-25-bert_tiny_finetuned_sst2_en

* Add model 2023-10-25-dk_emotion_bert_2_en

* Add model 2023-10-25-goog_bert_ft_cola_71_en

* Add model 2023-10-25-bert_base_uncased_finetuned_iemocap_uptuna2_en

* Add model 2023-10-25-goog_bert_ft_cola_73_en

* Add model 2023-10-25-bert_classification_5ksamples_en

* Add model 2023-10-25-bert_classifier_ara_multiclass_news_ar

* Add model 2023-10-25-tonely_bert_en

* Add model 2023-10-25-bert_base_uncased_sst_2_64_13_30_en

* Add model 2023-10-25-bert_ft_cola_81_en

* Add model 2023-10-25-bert_classification_10ksamples_en

* Add model 2023-10-25-fine_tuned_indonli_augmented_with_indobert_large_p2_muhammadravi251001_en

* Add model 2023-10-25-goog_bert_ft_cola_72_en

* Add model 2023-10-25-bert_ft_cola_82_en

* Add model 2023-10-25-bert_base_uncased_ag_news_finetuned_dwnews_categories_en

* Add model 2023-10-25-rubert_large_emotion_russian_cedr_m7_en

* Add model 2023-10-25-mengzi_bert_base_fin_ssec_en

* Add model 2023-10-25-ogbv_gender_bert_hindi_english_hasoc20a_fin_en

* Add model 2023-10-25-incivility_v2_final_tulio_chilean_spanish_bert_en

* Add model 2023-10-25-bert_ft_cola_84_en

* Add model 2023-10-25-goog_bert_ft_cola_63_en

* Add model 2023-10-25-goog_bert_ft_cola_44_en

* Add model 2023-10-25-goog_bert_ft_cola_14_en

* Add model 2023-10-25-hindi_const21_hibert_final_en

* Add model 2023-10-25-goog_bert_ft_cola_68_en

* Add model 2023-10-25-goog_bert_ft_cola_69_en

* Add model 2023-10-25-goog_bert_ft_cola_43_en

* Add model 2023-10-25-bert_large_uncased_sst_2_16_13_30_en

* Add model 2023-10-25-bert_classifier_bert_base_german_cased_gnad10_de

* Add model 2023-10-25-bert_finetuned_mrpc_tiansiyuan_en

* Add model 2023-10-25-bert_ft_cola_87_en

* Add model 2023-10-25-goog_bert_ft_cola_52_en

* Add model 2023-10-25-bert_tiny_finetuned_qqp_en

* Add model 2023-10-25-goog_bert_ft_cola_53_en

* Add model 2023-10-25-bert_base_uncased_sst_2_16_100_en

* Add model 2023-10-25-bert_ft_cola_88_en

* Add model 2023-10-25-goog_bert_ft_cola_39_en

* Add model 2023-10-25-goog_bert_ft_cola_60_en

* Add model 2023-10-25-bert_mixed_en

* Add model 2023-10-25-bert_ft_qqp_83_jeevesh8_en

* Add model 2023-10-25-bert_base_uncased_finetuned_question_v_statement_en

* Add model 2023-10-25-bert_finetuning_test_milian_en

* Add model 2023-10-25-baseline_bert_large_cased_epoch3_batch4_lr2e_05_w0_01_en

* Add model 2023-10-25-goog_bert_ft_cola_36_en

* Add model 2023-10-25-goog_bert_ft_cola_48_en

* Add model 2023-10-25-bert_ft_qqp_97_jeevesh8_en

* Add model 2023-10-25-bert_ft_qqp_35_jeevesh8_en

* Add model 2023-10-25-goog_bert_ft_cola_19_en

* Add model 2023-10-25-bert_finetunning_test_jovenpai_en

* Add model 2023-10-25-bert_ft_cola_89_en

* Add model 2023-10-25-bert_ft_cola_83_en

* Add model 2023-10-25-goog_bert_ft_cola_75_en

* Add model 2023-10-25-bert_sequence_classifier_base_uncased_sst2_en

* Add model 2023-10-25-bert_large_uncased_sst_2_64_13_30_en

* Add model 2023-10-25-tiny_bert_mrpc_distilled_en

* Add model 2023-10-25-fine_tuned_indonli_translated_with_indobert_base_uncased_muhammadravi251001_en

* Add model 2023-10-25-bert_ft_qqp_64_jeevesh8_en

* Add model 2023-10-25-bert_ft_qqp_31_jeevesh8_en

* Add model 2023-10-25-bert_base_arabic_camelbert_msa_sixteenth_xnli_finetuned_sarahadnan_en

* Add model 2023-10-25-goog_bert_ft_cola_56_en

* Add model 2023-10-25-std_0pnt2_bert_ft_cola_16_en

* Add model 2023-10-25-bert_mini_finetuned_mnli_en

* Add model 2023-10-25-bert_base_uncased_mnli_sparse_70_unstructured_en

* Add model 2023-10-25-bert_ft_cola_90_en

* Add model 2023-10-25-finetuned_bert_mrpc_ndugar_en

* Add model 2023-10-25-bert_ft_qqp_92_jeevesh8_en

* Add model 2023-10-25-bert_ft_qqp_88_jeevesh8_en

* Add model 2023-10-25-solved_finbert_tone_en

* Add model 2023-10-25-deprem_berturk_binary_tr

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_open_ssec_f1_v2_en

* Add model 2023-10-25-bertabaporu_portuguese_irony_en

* Add model 2023-10-25-bert_sequence_classifier_persian_farsi_base_uncased_sentiment_snappfood_fa

* Add model 2023-10-25-goog_bert_ft_cola_51_en

* Add model 2023-10-25-bert_base_uncased_sst_2_32_13_en

* Add model 2023-10-25-fine_tuned_indonli_translated_with_indobert_large_p2_en

* Add model 2023-10-25-goog_bert_ft_cola_37_en

* Add model 2023-10-25-std_0pnt2_bert_ft_cola_62_en

* Add model 2023-10-25-bert_ft_cola_91_en

* Add model 2023-10-25-sentiment_hts2_hubert_hungarian_hu

* Add model 2023-10-25-bert_ft_qqp_36_jeevesh8_en

* Add model 2023-10-25-bert_wiki_comments_finetuned_en

* Add model 2023-10-25-norsk_bert_fintuned_en

* Add model 2023-10-25-bert_sim_pair_en

* Add model 2023-10-25-bert_ft_qqp_62_jeevesh8_en

* Add model 2023-10-25-bert_base_uncased_nisadibipolar_en

* Add model 2023-10-25-samyarn_bert_base_multilingual_cased_xx

* Add model 2023-10-25-bert_italian_sentiment_en

* Add model 2023-10-25-bert_ft_cola_92_en

* Add model 2023-10-25-fine_tuned_koreanindonli_kornli_with_bert_kor_base_en

* Add model 2023-10-25-bert_mini_finetuned_qnli_en

* Add model 2023-10-25-bert_ft_qqp_94_jeevesh8_en

* Add model 2023-10-25-sentiment_hts5_hubert_hungarian_hu

* Add model 2023-10-25-bert_ft_qqp_77_jeevesh8_en

* Add model 2023-10-25-bert_ft_qqp_38_jeevesh8_en

* Add model 2023-10-25-bert_tweets_semeval_unclean_en

* Add model 2023-10-25-bert_ft_cola_80_en

* Add model 2023-10-25-goog_bert_ft_cola_30_en

* Add model 2023-10-25-bert_ft_qqp_75_jeevesh8_en

* Add model 2023-10-25-fine_tuned_indonli_basic_with_indobert_base_uncased_afaji_en

* Add model 2023-10-25-esg_bert_sector_classifier_en

* Add model 2023-10-25-bert_mini_finetuned_stsb_en

* Add model 2023-10-25-bert_large_uncased_sst_2_16_13_smoothed_en

* Add model 2023-10-25-goog_bert_ft_cola_24_en

* Add model 2023-10-25-bert_ft_cola_93_en

* Add model 2023-10-25-bert_ft_qqp_90_jeevesh8_en

* Add model 2023-10-25-bert_mini_finetuned_qqp_en

* Add model 2023-10-25-goog_bert_ft_cola_33_en

* Add model 2023-10-25-fine_tuned_indonli_augmented_with_indobert_large_p2_afaji_en

* Add model 2023-10-25-bert_ft_cola_85_en

* Add model 2023-10-25-bert_large_uncased_hoax_classifier_v1_en

* Add model 2023-10-25-roberta_fake_real_en

* Add model 2023-10-25-bert_tiny_finetuned_stsb_en

* Add model 2023-10-25-bert_ft_qqp_86_jeevesh8_en

* Add model 2023-10-25-bert_sequence_classifier_persian_farsi_base_uncased_sentiment_deepsentipers_multi_fa

* Add model 2023-10-25-bert_base_uncased_reviews_4_en

* Add model 2023-10-25-goog_bert_ft_cola_55_en

* Add model 2023-10-25-fine_tuned_indonli_basic_with_indobert_large_p2_muhammadravi251001_en

* Add model 2023-10-25-bert_ft_cola_94_en

* Add model 2023-10-25-bert_large_uncased_sst_2_32_13_30_en

* Add model 2023-10-25-bert_ft_qqp_42_jeevesh8_en

* Add model 2023-10-25-bert_italian_irony_en

* Add model 2023-10-25-bert_large_uncased_cola_b_en

* Add model 2023-10-25-bert_ft_qqp_96_jeevesh8_en

* Add model 2023-10-25-bert_base_mdoc_bm25_en

* Add model 2023-10-25-bert_ft_cola_86_en

* Add model 2023-10-25-bert_ft_qqp_84_jeevesh8_en

* Add model 2023-10-25-goog_bert_ft_cola_9_en

* Add model 2023-10-25-bert_base_multilingual_cased_mrpc_glue_xx

* Add model 2023-10-25-goog_bert_ft_cola_54_en

* Add model 2023-10-25-bert_finetuned_winogrande_en

* Add model 2023-10-25-bert_fakenews_en

* Add model 2023-10-25-bert_ft_cola_95_en

* Add model 2023-10-25-goog_bert_ft_cola_57_en

* Add model 2023-10-25-fine_tuned_indonli_augmented_with_indobert_base_uncased_muhammadravi251001_en

* Add model 2023-10-25-bert_base_mdoc_hdct_en

* Add model 2023-10-25-bert_ft_qqp_98_jeevesh8_en

* Add model 2023-10-25-bert_sequce_classifier_paraphrase_en

* Add model 2023-10-25-mengzi_bert_base_fin_wallstreetcn_morning_news_market_overview_open_ssec_f1_v22_en

* Add model 2023-10-25-bert_sequence_classifier_persian_farsi_base_uncased_sentiment_digikala_fa

* Add model 2023-10-25-fine_tuned_indonli_basic_with_indobert_base_uncased_muhammadravi251001_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_1_en

* Add model 2023-10-25-goog_bert_ft_cola_40_en

* Add model 2023-10-25-goog_bert_ft_cola_50_en

* Add model 2023-10-25-goog_bert_ft_cola_12_en

* Add model 2023-10-25-bert_base_uncased_finetuned_hateful_meme_en

* Add model 2023-10-25-bert_ft_cola_96_en

* Add model 2023-10-25-bert_ft_qqp_72_jeevesh8_en

* Add model 2023-10-25-goog_bert_ft_cola_22_en

* Add model 2023-10-25-bert_wikipedia_sst2_en

* Add model 2023-10-25-goog_bert_ft_cola_26_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_2_en

* Add model 2023-10-25-bert_deepfake_bulgarian_multiclass_bg

* Add model 2023-10-25-bert_ft_qqp_52_jeevesh8_en

* Add model 2023-10-25-bert_ft_qqp_56_jeevesh8_en

* Add model 2023-10-25-goog_bert_ft_cola_49_en

* Add model 2023-10-25-goog_bert_ft_cola_64_en

* Add model 2023-10-25-indobertnewstest_rizalmilyardi_en

* Add model 2023-10-25-a01_suicide_bert_huggingface_finetune_en

* Add model 2023-10-25-bert_base_arabic_camelbert_msa_finetuned_arabic_dialect_identification_model_1_ar

* Add model 2023-10-25-bert_base_uncased_sst_2_16_13_30_en

* Add model 2023-10-25-bert_ft_cola_97_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_3_en

* Add model 2023-10-25-bert_ft_qqp_45_jeevesh8_en

* Add model 2023-10-25-goog_bert_ft_cola_32_en

* Add model 2023-10-25-goog_bert_ft_cola_74_en

* Add model 2023-10-25-norbert2_sentiment_norec_2_en

* Add model 2023-10-25-biobert_finetuned_genetic_mutation_en

* Add model 2023-10-25-bert_sequence_classifier_paraphrase_pt

* Add model 2023-10-25-bert_sim_doc_en

* Add model 2023-10-25-goog_bert_ft_cola_70_en

* Add model 2023-10-25-goog_bert_ft_cola_42_en

* Add model 2023-10-25-bert_finetuning_test_itcastai_en

* Add model 2023-10-25-bert_ft_cola_98_en

* Add model 2023-10-25-goog_bert_ft_cola_18_en

* Add model 2023-10-25-baseline_bert_base_cased_epoch3_batch4_lr2e_05_w0_01_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_4_en

* Add model 2023-10-25-bert_ft_qqp_99_jeevesh8_en

* Add model 2023-10-25-bert_classifier_dehate_mono_indonesian_en

* Add model 2023-10-25-bert_italian_hate_speech_en

* Add model 2023-10-25-bert_classifier_base_uncased_qnli_en

* Add model 2023-10-25-goog_bert_ft_cola_21_en

* Add model 2023-10-25-goog_bert_ft_cola_31_en

* Add model 2023-10-25-goog_bert_ft_cola_67_en

* Add model 2023-10-25-a02_suicide_bert_huggingface_finetune_en

* Add model 2023-10-25-bert_classifier_finbert_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_5_en

* Add model 2023-10-25-bert_ft_cola_99_en

* Add model 2023-10-25-bert_ft_qqp_82_jeevesh8_en

* Add model 2023-10-25-goog_bert_ft_cola_3_en

* Add model 2023-10-25-bert_base_chinese_finetuned_binary_best_en

* Add model 2023-10-25-bert_classification_1500samples_en

* Add model 2023-10-25-goog_bert_ft_cola_35_en

* Add model 2023-10-25-tiny_bert_sst2_distilled_kushaljoseph_en

* Add model 2023-10-25-goog_bert_ft_cola_65_en

* Add model 2023-10-25-chinese_roberta_wwm_ext_finetuned_binary_en

* Add model 2023-10-25-goog_bert_ft_cola_58_en

* Add model 2023-10-25-norbert2_sentiment_norec_4_en

* Add model 2023-10-25-bert_classifier_russian_base_srl_ru

* Add model 2023-10-25-goog_bert_ft_cola_59_en

* Add model 2023-10-25-covid_twitter_bert_v2_mnli_nli_sts_crossencoder_covid_hera_en

* Add model 2023-10-25-norbert2_sentiment_norec_6_en

* Add model 2023-10-25-legal_bert_tpb_clause_class_en

* Add model 2023-10-25-goog_bert_ft_cola_61_en

* Add model 2023-10-25-twitter_disaster_bert_large_en

* Add model 2023-10-25-goog_bert_ft_cola_62_en

* Add model 2023-10-25-bert_large_uncased_mnli_ofirzaf_en

* Add model 2023-10-25-norbert2_sentiment_norec_7_en

* Add model 2023-10-25-bert_classifier_riad_finetuned_mrpc_en

* Add model 2023-10-25-bert_base_cased_finetuned_cola_sreyang_nvidia_en

* Add model 2023-10-25-bert_base_uncased_research_articles_multilabel_en

* Add model 2023-10-25-goog_bert_ft_cola_66_en

* Add model 2023-10-25-mini_bert_distilled_en

* Add model 2023-10-25-bert_fom_job_description_assignment_en

* Add model 2023-10-25-goog_bert_ft_cola_76_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_6_en

* Add model 2023-10-25-finbert_finetuned_fg_single_sentence_news_weighted_en

* Add model 2023-10-25-bert_cl_cf_1700_en

* Add model 2023-10-25-goog_bert_ft_cola_77_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_7_en

* Add model 2023-10-25-bert_classifier_prot_bfd_localization_en

* Add model 2023-10-25-bert_cl_g_1700_en

* Add model 2023-10-25-bert_sentence_classifier_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_8_en

* Add model 2023-10-25-goog_bert_ft_cola_81_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_9_en

* Add model 2023-10-25-rubert_rusentitweet_sismetanin_en

* Add model 2023-10-25-goog_bert_ft_cola_91_en

* Add model 2023-10-25-norbert2_sentiment_norec_8_en

* Add model 2023-10-25-bert_base_chinese_wallstreetcn_morning_news_market_overview_ssec_10_en

* Add model 2023-10-25-goog_bert_ft_cola_89_en

* Add model 2023-10-25-bert_base_uncased_random_weights_s42_en

* Add model 2023-10-25-norbert2_sentiment_norec_9_en

* Add model 2023-10-25-bert_base_uncased_finetuned_cola_ruizhou_en

* Add model 2023-10-25-fine_tuned_koreanindonli_kornli_with_bert_base_en

* Add model 2023-10-25-goog_bert_ft_cola_82_en

* Add model 2023-10-25-finbert_fls_en

* Add model 2023-10-25-bert_base_uncased_finetuned_mrpc_ruizhou_en

* Add model 2023-10-25-bert_cnn_news_en

---------

Co-authored-by: ahmedlone127 <ahmedlone127@gmail.com>
  • Loading branch information
jsl-models and ahmedlone127 authored Oct 25, 2023
1 parent b505d77 commit 824a221
Show file tree
Hide file tree
Showing 301 changed files with 29,288 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
---
layout: model
title: English a01_suicide_bert_huggingface_finetune BertForSequenceClassification from Ariffnaz
author: John Snow Labs
name: a01_suicide_bert_huggingface_finetune
date: 2023-10-25
tags: [bert, en, open_source, sequence_classification, onnx]
task: Text Classification
language: en
edition: Spark NLP 5.1.4
spark_version: 3.4
supported: true
engine: onnx
annotator: BertForSequenceClassification
article_header:
type: cover
use_language_switcher: "Python-Scala-Java"
---

## Description

Pretrained BertForSequenceClassification model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.`a01_suicide_bert_huggingface_finetune` is a English model originally trained by Ariffnaz.

{:.btn-box}
<button class="button button-orange" disabled>Live Demo</button>
<button class="button button-orange" disabled>Open in Colab</button>
[Download](https://s3.amazonaws.com/auxdata.johnsnowlabs.com/public/models/a01_suicide_bert_huggingface_finetune_en_5.1.4_3.4_1698231600777.zip){:.button.button-orange.button-orange-trans.arr.button-icon}
[Copy S3 URI](s3://auxdata.johnsnowlabs.com/public/models/a01_suicide_bert_huggingface_finetune_en_5.1.4_3.4_1698231600777.zip){:.button.button-orange.button-orange-trans.button-icon.button-copy-s3}

## How to use



<div class="tabs-box" markdown="1">
{% include programmingLanguageSelectScalaPythonNLU.html %}
```python

document_assembler = DocumentAssembler()\
.setInputCol("text")\
.setOutputCol("document")

tokenizer = Tokenizer()\
.setInputCols("document")\
.setOutputCol("token")

sequenceClassifier = BertForSequenceClassification.pretrained("a01_suicide_bert_huggingface_finetune","en")\
.setInputCols(["document","token"])\
.setOutputCol("class")

pipeline = Pipeline().setStages([document_assembler, tokenizer, sequenceClassifier])

data = spark.createDataFrame([["PUT YOUR STRING HERE"]]).toDF("text")

result = pipeline.fit(data).transform(data)

```
```scala

val document_assembler = new DocumentAssembler()
.setInputCol("text")
.setOutputCol("document")

val tokenizer = new Tokenizer()
.setInputCols("document")
.setOutputCol("token")

val sequenceClassifier = BertForSequenceClassification.pretrained("a01_suicide_bert_huggingface_finetune","en")
.setInputCols(Array("document","token"))
.setOutputCol("class")

val pipeline = new Pipeline().setStages(Array(documentAssembler, tokenizer, sequenceClassifier))

val data = Seq("PUT YOUR STRING HERE").toDS.toDF("text")

val result = pipeline.fit(data).transform(data)


```
</div>

{:.model-param}
## Model Information

{:.table-model}
|---|---|
|Model Name:|a01_suicide_bert_huggingface_finetune|
|Compatibility:|Spark NLP 5.1.4+|
|License:|Open Source|
|Edition:|Official|
|Input Labels:|[documents, token]|
|Output Labels:|[class]|
|Language:|en|
|Size:|409.4 MB|

## References

https://huggingface.co/Ariffnaz/a01-suicide-bert-huggingface-finetune
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
---
layout: model
title: English a02_suicide_bert_huggingface_finetune BertForSequenceClassification from Ariffnaz
author: John Snow Labs
name: a02_suicide_bert_huggingface_finetune
date: 2023-10-25
tags: [bert, en, open_source, sequence_classification, onnx]
task: Text Classification
language: en
edition: Spark NLP 5.1.4
spark_version: 3.4
supported: true
engine: onnx
annotator: BertForSequenceClassification
article_header:
type: cover
use_language_switcher: "Python-Scala-Java"
---

## Description

Pretrained BertForSequenceClassification model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.`a02_suicide_bert_huggingface_finetune` is a English model originally trained by Ariffnaz.

{:.btn-box}
<button class="button button-orange" disabled>Live Demo</button>
<button class="button button-orange" disabled>Open in Colab</button>
[Download](https://s3.amazonaws.com/auxdata.johnsnowlabs.com/public/models/a02_suicide_bert_huggingface_finetune_en_5.1.4_3.4_1698233704009.zip){:.button.button-orange.button-orange-trans.arr.button-icon}
[Copy S3 URI](s3://auxdata.johnsnowlabs.com/public/models/a02_suicide_bert_huggingface_finetune_en_5.1.4_3.4_1698233704009.zip){:.button.button-orange.button-orange-trans.button-icon.button-copy-s3}

## How to use



<div class="tabs-box" markdown="1">
{% include programmingLanguageSelectScalaPythonNLU.html %}
```python

document_assembler = DocumentAssembler()\
.setInputCol("text")\
.setOutputCol("document")

tokenizer = Tokenizer()\
.setInputCols("document")\
.setOutputCol("token")

sequenceClassifier = BertForSequenceClassification.pretrained("a02_suicide_bert_huggingface_finetune","en")\
.setInputCols(["document","token"])\
.setOutputCol("class")

pipeline = Pipeline().setStages([document_assembler, tokenizer, sequenceClassifier])

data = spark.createDataFrame([["PUT YOUR STRING HERE"]]).toDF("text")

result = pipeline.fit(data).transform(data)

```
```scala

val document_assembler = new DocumentAssembler()
.setInputCol("text")
.setOutputCol("document")

val tokenizer = new Tokenizer()
.setInputCols("document")
.setOutputCol("token")

val sequenceClassifier = BertForSequenceClassification.pretrained("a02_suicide_bert_huggingface_finetune","en")
.setInputCols(Array("document","token"))
.setOutputCol("class")

val pipeline = new Pipeline().setStages(Array(documentAssembler, tokenizer, sequenceClassifier))

val data = Seq("PUT YOUR STRING HERE").toDS.toDF("text")

val result = pipeline.fit(data).transform(data)


```
</div>

{:.model-param}
## Model Information

{:.table-model}
|---|---|
|Model Name:|a02_suicide_bert_huggingface_finetune|
|Compatibility:|Spark NLP 5.1.4+|
|License:|Open Source|
|Edition:|Official|
|Input Labels:|[documents, token]|
|Output Labels:|[class]|
|Language:|en|
|Size:|409.4 MB|

## References

https://huggingface.co/Ariffnaz/a02-suicide-bert-huggingface-finetune
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
---
layout: model
title: English ag_news_bert_base_uncased BertForSequenceClassification from Kyle1668
author: John Snow Labs
name: ag_news_bert_base_uncased
date: 2023-10-25
tags: [bert, en, open_source, sequence_classification, onnx]
task: Text Classification
language: en
edition: Spark NLP 5.1.4
spark_version: 3.4
supported: true
engine: onnx
annotator: BertForSequenceClassification
article_header:
type: cover
use_language_switcher: "Python-Scala-Java"
---

## Description

Pretrained BertForSequenceClassification model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.`ag_news_bert_base_uncased` is a English model originally trained by Kyle1668.

{:.btn-box}
<button class="button button-orange" disabled>Live Demo</button>
<button class="button button-orange" disabled>Open in Colab</button>
[Download](https://s3.amazonaws.com/auxdata.johnsnowlabs.com/public/models/ag_news_bert_base_uncased_en_5.1.4_3.4_1698221722413.zip){:.button.button-orange.button-orange-trans.arr.button-icon}
[Copy S3 URI](s3://auxdata.johnsnowlabs.com/public/models/ag_news_bert_base_uncased_en_5.1.4_3.4_1698221722413.zip){:.button.button-orange.button-orange-trans.button-icon.button-copy-s3}

## How to use



<div class="tabs-box" markdown="1">
{% include programmingLanguageSelectScalaPythonNLU.html %}
```python

document_assembler = DocumentAssembler()\
.setInputCol("text")\
.setOutputCol("document")

tokenizer = Tokenizer()\
.setInputCols("document")\
.setOutputCol("token")

sequenceClassifier = BertForSequenceClassification.pretrained("ag_news_bert_base_uncased","en")\
.setInputCols(["document","token"])\
.setOutputCol("class")

pipeline = Pipeline().setStages([document_assembler, tokenizer, sequenceClassifier])

data = spark.createDataFrame([["PUT YOUR STRING HERE"]]).toDF("text")

result = pipeline.fit(data).transform(data)

```
```scala

val document_assembler = new DocumentAssembler()
.setInputCol("text")
.setOutputCol("document")

val tokenizer = new Tokenizer()
.setInputCols("document")
.setOutputCol("token")

val sequenceClassifier = BertForSequenceClassification.pretrained("ag_news_bert_base_uncased","en")
.setInputCols(Array("document","token"))
.setOutputCol("class")

val pipeline = new Pipeline().setStages(Array(documentAssembler, tokenizer, sequenceClassifier))

val data = Seq("PUT YOUR STRING HERE").toDS.toDF("text")

val result = pipeline.fit(data).transform(data)


```
</div>

{:.model-param}
## Model Information

{:.table-model}
|---|---|
|Model Name:|ag_news_bert_base_uncased|
|Compatibility:|Spark NLP 5.1.4+|
|License:|Open Source|
|Edition:|Official|
|Input Labels:|[documents, token]|
|Output Labels:|[class]|
|Language:|en|
|Size:|409.4 MB|

## References

https://huggingface.co/Kyle1668/ag-news-bert-base-uncased
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
---
layout: model
title: English autotrain_bert_wikipedia_sst_2_1034235513 BertForSequenceClassification from deepesh0x
author: John Snow Labs
name: autotrain_bert_wikipedia_sst_2_1034235513
date: 2023-10-25
tags: [bert, en, open_source, sequence_classification, onnx]
task: Text Classification
language: en
edition: Spark NLP 5.1.4
spark_version: 3.4
supported: true
engine: onnx
annotator: BertForSequenceClassification
article_header:
type: cover
use_language_switcher: "Python-Scala-Java"
---

## Description

Pretrained BertForSequenceClassification model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.`autotrain_bert_wikipedia_sst_2_1034235513` is a English model originally trained by deepesh0x.

{:.btn-box}
<button class="button button-orange" disabled>Live Demo</button>
<button class="button button-orange" disabled>Open in Colab</button>
[Download](https://s3.amazonaws.com/auxdata.johnsnowlabs.com/public/models/autotrain_bert_wikipedia_sst_2_1034235513_en_5.1.4_3.4_1698212254160.zip){:.button.button-orange.button-orange-trans.arr.button-icon}
[Copy S3 URI](s3://auxdata.johnsnowlabs.com/public/models/autotrain_bert_wikipedia_sst_2_1034235513_en_5.1.4_3.4_1698212254160.zip){:.button.button-orange.button-orange-trans.button-icon.button-copy-s3}

## How to use



<div class="tabs-box" markdown="1">
{% include programmingLanguageSelectScalaPythonNLU.html %}
```python

document_assembler = DocumentAssembler()\
.setInputCol("text")\
.setOutputCol("document")

tokenizer = Tokenizer()\
.setInputCols("document")\
.setOutputCol("token")

sequenceClassifier = BertForSequenceClassification.pretrained("autotrain_bert_wikipedia_sst_2_1034235513","en")\
.setInputCols(["document","token"])\
.setOutputCol("class")

pipeline = Pipeline().setStages([document_assembler, tokenizer, sequenceClassifier])

data = spark.createDataFrame([["PUT YOUR STRING HERE"]]).toDF("text")

result = pipeline.fit(data).transform(data)

```
```scala

val document_assembler = new DocumentAssembler()
.setInputCol("text")
.setOutputCol("document")

val tokenizer = new Tokenizer()
.setInputCols("document")
.setOutputCol("token")

val sequenceClassifier = BertForSequenceClassification.pretrained("autotrain_bert_wikipedia_sst_2_1034235513","en")
.setInputCols(Array("document","token"))
.setOutputCol("class")

val pipeline = new Pipeline().setStages(Array(documentAssembler, tokenizer, sequenceClassifier))

val data = Seq("PUT YOUR STRING HERE").toDS.toDF("text")

val result = pipeline.fit(data).transform(data)


```
</div>

{:.model-param}
## Model Information

{:.table-model}
|---|---|
|Model Name:|autotrain_bert_wikipedia_sst_2_1034235513|
|Compatibility:|Spark NLP 5.1.4+|
|License:|Open Source|
|Edition:|Official|
|Input Labels:|[documents, token]|
|Output Labels:|[class]|
|Language:|en|
|Size:|409.4 MB|

## References

https://huggingface.co/deepesh0x/autotrain-bert_wikipedia_sst_2-1034235513
Loading

0 comments on commit 824a221

Please sign in to comment.