train(train_file,test_file,vocab_intent_file,vocab_slot_file)
Project description
Intent and Slot Filling Bi-Model
train(train_file = "data/train_dev", test_file = "data/test", vocab_intent_file = "data/vocab.intent", vocab_slot_file = "data/vocab.slot", device = torch.device("cuda" if torch.cuda.is_available() else "cpu"), total_epoch = 50, max_len = 50, batch = 16, learning_rate = 0.001, DROPOUT = 0.2, embedding_size = 300, lstm_hidden_size = 200)
5 files would be generated after training: model_intent_best.pt model_slot_best.pt word_dict.pickle slot_dict.pickle intent_dict.pickle
predict(sentence="find nonstop flights from salt lake city to new york on saturday april ninth.", data_path="/Users/chenzichu/Desktop/slot-filling/package/slot_ZC", device = torch.device("cuda" if torch.cuda.is_available() else "cpu"), max_len = 50, batch = 16, DROPOUT = 0.2, embedding_size = 300, lstm_hidden_size = 200)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for slot_filling_ZC-0.2.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 98712bd9f62283e427ece7d27c938e5e5613c7802f272bbd92d0037ae0ffe494 |
|
MD5 | 3d4c0c87f2e5d1176661f96b67bb8e5e |
|
BLAKE2b-256 | 761c05ada66ec6453a1d59ec9422208f77ec8468a4497f2db9996ce3220cd878 |