Kartik17 commited on
Commit
12e887d
1 Parent(s): a4c4ca7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,14 +5,14 @@ import pandas as pd
5
  from tensorflow.keras.layers import TextVectorization
6
 
7
  # Load your data
8
- new_data = pd.read_csv('data_train.csv') # Make sure to adjust the path if necessary
9
 
10
  x = new_data['comment_text']
11
  y = new_data[new_data.columns[2:]].values
12
 
13
- max_features = 100000
14
  vectorizer = TextVectorization(max_tokens=max_features,
15
- output_sequence_length=1500,
16
  output_mode='int')
17
  vectorizer.get_vocabulary()
18
  vectorizer.adapt(x.values)
 
5
  from tensorflow.keras.layers import TextVectorization
6
 
7
  # Load your data
8
+ new_data = pd.read_csv('train.csv') # Make sure to adjust the path if necessary
9
 
10
  x = new_data['comment_text']
11
  y = new_data[new_data.columns[2:]].values
12
 
13
+ max_features = 160000
14
  vectorizer = TextVectorization(max_tokens=max_features,
15
+ output_sequence_length=1800,
16
  output_mode='int')
17
  vectorizer.get_vocabulary()
18
  vectorizer.adapt(x.values)