uservipin commited on
Commit
da84cac
1 Parent(s): 2286df4

Auc roc curve

Browse files
Files changed (4) hide show
  1. NLP.py +16 -0
  2. ROC.png +0 -0
  3. __pycache__/regression.cpython-310.pyc +0 -0
  4. requirements.txt +5 -1
NLP.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use a pipeline as a high-level helper
2
+ from transformers import pipeline
3
+ print("Hello world")
4
+ pipe = pipeline("translation", model="facebook/nllb-200-distilled-600M")
5
+
6
+ from transformers import AutoTokenizer, M2M100ForConditionalGeneration
7
+
8
+ model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
9
+ tokenizer = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
10
+
11
+ text_to_translate = "Life is like a box of chocolates"
12
+ model_inputs = tokenizer(text_to_translate, return_tensors="pt")
13
+
14
+ # translate to French
15
+ gen_tokens = model.generate(**model_inputs, forced_bos_token_id=tokenizer.get_lang_id("fr"))
16
+ print(tokenizer.batch_decode(gen_tokens, skip_special_tokens=True))
ROC.png ADDED
__pycache__/regression.cpython-310.pyc CHANGED
Binary files a/__pycache__/regression.cpython-310.pyc and b/__pycache__/regression.cpython-310.pyc differ
 
requirements.txt CHANGED
@@ -1,4 +1,8 @@
1
- matplotlib==3.8.3
2
  pandas==1.5.3
 
3
  scikit_learn==1.4.1.post1
4
  streamlit==1.32.0
 
 
 
 
1
+ lightgbm==4.3.0
2
  pandas==1.5.3
3
+ scikit_learn==1.2.1
4
  scikit_learn==1.4.1.post1
5
  streamlit==1.32.0
6
+ transformers==4.24.0
7
+ transformers==4.39.2
8
+ xgboost==2.0.3