Skip to main content

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

slot-filling-ZC-0.2.4.tar.gz (10.4 kB view hashes)

Uploaded source

Built Distribution

slot_filling_ZC-0.2.4-py3-none-any.whl (130.4 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page