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.1.9-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c1f30968e8666f1dfaffcdf2250ecb284c1cf978fa6bebad31c451e54919ac34 |
|
MD5 | 8dfe71c403eee5dbcfa6b3b666c91164 |
|
BLAKE2b-256 | 77b5549d64e493f0301902361f20eaf075b0fb4c8937d527fc40052b1f6a9344 |