beimingwu
/
learnware
Not watched
1
0
Fork 0
Code
Releases 1 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: f281e3efed
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'f281e3efed'
${ noResults }
learnware/examples/dataset_text_workflow2/get_data.py

12 lines
369 B
Raw Blame History 

  1. import os

  2. import json

  3. import numpy as np

  4. from sklearn.datasets import fetch_20newsgroups

  5. import pandas as pd

  6. 

  7. 

  8. def get_data(data_root):

  9.     X_train, y_train = fetch_20newsgroups(data_home=data_root, subset='train', return_X_y=True)

  10.     X_test, y_test = fetch_20newsgroups(data_home=data_root, subset='test', return_X_y=True)

  11. 

  12.     return X_train, y_train, X_test, y_test