Spaces:
Sleeping
Sleeping
Rename constructor.py to chatbot_constructor.py
Browse files
constructor.py → chatbot_constructor.py
RENAMED
@@ -15,7 +15,7 @@ os.mkdir("cache")
|
|
15 |
def hash_str(data: str):
|
16 |
return hashlib.md5(data.encode('utf-8')).hexdigest()
|
17 |
|
18 |
-
def train(message: str, epochs: int, learning_rate: float, emb_size: int, inp_len: int, data: str):
|
19 |
if "→" not in data or "\n" not in data:
|
20 |
return "Dataset example:\nquestion→answer\nquestion→answer\netc."
|
21 |
dset, responses = todset(data)
|
|
|
15 |
def hash_str(data: str):
|
16 |
return hashlib.md5(data.encode('utf-8')).hexdigest()
|
17 |
|
18 |
+
def train(message: str = "", epochs: int = 16, learning_rate: float = 0.001, emb_size: int = 128, inp_len: int = 16, data: str = ""):
|
19 |
if "→" not in data or "\n" not in data:
|
20 |
return "Dataset example:\nquestion→answer\nquestion→answer\netc."
|
21 |
dset, responses = todset(data)
|