Datasets ⭐
ATIS Dataset
Load ATIS (Airline Travel Information Systems) dataset in Python fast. Intent classification dataset. Stream ATIS while training ML models PyTorch & TensorFlow.

ATIS dataset

What is ATIS Dataset?

The ATIS (Airline Travel Information Systems) dataset contains audio recordings and hand transcripts of individuals querying automated airline travel inquiry systems for flight information. There are 17 distinct purpose types in the data. In the train, development, and test sets, there are 4478, 500, and 893 intent-labeled reference utterances, respectively.

Download ATIS Dataset in Python

Instead of downloading the ATIS dataset in Python, you can effortlessly load it in Python via our open-source package Hub with just one line of code.

Load ATIS Dataset Training Subset in Python

import hub
ds = hub.load("hub://activeloop/atis-train")

Load ATIS Dataset Testing Subset in Python

import hub
ds = hub.load("hub://activeloop/atis-test")

ATIS Dataset Structure

ATIS Data Fields

  • intent : tensor containing the labels that represent intents.
  • entity : tensor containing entity
  • end : tensor contating index of the end of the sentence
  • start: tensor containing the index of the beginning of the sentence.
  • value: tensor containing value.
  • text: tensor containing the text.

ATIS Data Splits

  • The ATIS dataset training set is composed of 14923 samples and 22 classes.
  • The ATIS dataset testing set is composed of 2677 samples and 20 classes.

How to use ATIS Dataset with PyTorch and TensorFlow in Python

Train a model on ATIS dataset with PyTorch in Python

Let's use Hub's built-in PyTorch one-line dataloader to connect the data to the compute:
dataloader = ds.pytorch(num_workers=0, batch_size=4, shuffle=False)

Train a model on ATIS dataset with TensorFlow in Python

dataloader = ds.tensorflow()

Additional Information about ATIS Dataset

ATIS Dataset Description

ATIS Dataset Curators

Prashanth Gurunath Shivakumar, Mu Yang, Panayiotis Georgiou

ATIS Dataset Licensing Information

Hub users may have access to a variety of publicly available datasets. We do not host or distribute these datasets, vouch for their quality or fairness, or claim that you have a license to use the datasets. It is your responsibility to determine whether you have permission to use the datasets under their license.
If you're a dataset owner and do not want your dataset to be included in this library, please get in touch through a GitHub issue. Thank you for your contribution to the ML community!

ATIS Dataset Citation Information

title={Spoken language intent detection using confusion2vec},
author={Shivakumar, Prashanth Gurunath and Yang, Mu and Georgiou, Panayiotis},
journal={arXiv preprint arXiv:1904.03576},

ATIS Dataset FAQs

What is the ATIS dataset for Python?

The ATIS (Airline Travel Information Systems) is a dataset consisting of audio recordings and corresponding manual transcripts about humans asking for flight information on automated airline travel inquiry systems.
How to download the ATIS dataset in Python?
You can load ATIS dataset fast with one line of code using the open-source package Activeloop Hub in Python. See detailed instructions on how to load ATIS dataset training subset or ATIS dataset testing subset in Python.