Bnava13 commited on
Commit
34e07b5
·
verified ·
1 Parent(s): e0d9668

Updated Features

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,11 +11,11 @@ data = pd.read_csv('steam.csv', quotechar='"', on_bad_lines='skip', nrows=10000)
11
  data.fillna('', inplace=True)
12
 
13
  # Combine features
14
- selected_features = ['genres', 'categories', 'tags']
15
  for feature in selected_features:
16
  if feature not in data.columns:
17
  data[feature] = ''
18
- data['combined_features'] = data['genres'] + ' ' + data['categories'] + ' ' + data['tags']
19
 
20
  # Vectorize
21
  vectorizer = TfidfVectorizer(stop_words='english', ngram_range=(1, 2), max_features=8000)
 
11
  data.fillna('', inplace=True)
12
 
13
  # Combine features
14
+ selected_features = ['genres', 'categories', 'steamspy_tags', 'platforms']
15
  for feature in selected_features:
16
  if feature not in data.columns:
17
  data[feature] = ''
18
+ data['combined_features'] = data['genres'] + ' ' + data['categories'] + ' ' + data['steamspy_tags'] + ' ' + data['platform']
19
 
20
  # Vectorize
21
  vectorizer = TfidfVectorizer(stop_words='english', ngram_range=(1, 2), max_features=8000)