ierhon commited on
Commit
a282150
1 Parent(s): 16a0d94

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)