modelData.py 文件源码

def splitData(features, labels, testSize = 0.3):
    '''
    Split data into train and test sets
    @param features: Features generated from data
    @param labels: symptom severity label for each note
    @param testSize: fraction of data to use for testing models
    @return feats_train: the features for training
    @return feats_test: the features for testing
    @return labels_train: symptom severity labels corresponding to training features
    @return labels_test: symptom severity labels corresponding to test features 
    '''

    feats_train, feats_test, labels_train, labels_test = cross_validation.train_test_split(features, labels, test_size=testSize, random_state=15)

    return(feats_train, feats_test, labels_train, labels_test)