Skip to content

Instantly share code, notes, and snippets.

@woobe

woobe/split_airline_sentiment.py

Created September 11, 2018 18:25
Download ZIP

Revisions

  1. woobe created this gist Sep 11, 2018.
    8 changes: 8 additions & 0 deletions split_airline_sentiment.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,8 @@
    import numpy as np
    import pandas as pd
    from sklearn import model_selection

    al = pd.read_csv("Airline-Sentiment-2-w-AA.csv", encoding='ISO-8859-1')
    train_al, test_al = model_selection.train_test_split(al, test_size=0.2, random_state=2018)
    train_al.to_csv("train_airline_sentiment.csv", index=False)
    test_al.to_csv("test_airline_sentiment.csv", index=False)