Spaces:
Sleeping
Sleeping
This works
Browse files- __pycache__/poketype.cpython-310.pyc +0 -0
- models/model.pickle +0 -0
- models/tfidf.pickle +0 -0
- poketype.py +1 -0
__pycache__/poketype.cpython-310.pyc
ADDED
|
Binary file (2.31 kB). View file
|
|
|
models/model.pickle
DELETED
|
Binary file (705 kB)
|
|
|
models/tfidf.pickle
DELETED
|
Binary file (352 kB)
|
|
|
poketype.py
CHANGED
|
@@ -41,6 +41,7 @@ class PokemonTypeIdentifier():
|
|
| 41 |
encoded_labels,decoded_labels = pd.factorize(df_majority['Type 1'])
|
| 42 |
self.model = LinearSVC().fit(features,encoded_labels)
|
| 43 |
self.decoded_labels = decoded_labels
|
|
|
|
| 44 |
if not self.isModelLoaded or not self.isFileFound:
|
| 45 |
raise AttributeError("Required File Doesn't Exist.")
|
| 46 |
def predict_type(self,poke_str):
|
|
|
|
| 41 |
encoded_labels,decoded_labels = pd.factorize(df_majority['Type 1'])
|
| 42 |
self.model = LinearSVC().fit(features,encoded_labels)
|
| 43 |
self.decoded_labels = decoded_labels
|
| 44 |
+
self.isModelLoaded = True
|
| 45 |
if not self.isModelLoaded or not self.isFileFound:
|
| 46 |
raise AttributeError("Required File Doesn't Exist.")
|
| 47 |
def predict_type(self,poke_str):
|