Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีใช้ Keras ดาวน์โหลดและสำรวจ Dataset StackOverflow สำหรับ Tag Prediction (ฉบับสมบูรณ์)

บทนำ

Keras High-level API รันบน TensorFlow รองรับ Cross-platform (TPU, GPU, Mobile, Web) มี Dataset Built-in และ Utilities ดาวน์โหลด

การติดตั้ง

pip install tensorflow
pip install tensorflow-text

ตัวอย่างโค้ด

import collections, pathlib, re, string
import tensorflow as tf
from tensorflow.keras import layers, losses, preprocessing, utils
from tensorflow.keras.layers.experimental.preprocessing import TextVectorization
import tensorflow_datasets as tfds
import tensorflow_text as tf_text

data_url = 'https://storage.googleapis.com/download.tensorflow.org/data/stack_overflow_16k.tar.gz'
dataset = utils.get_file(
    'stack_overflow_16k.tar.gz',
    data_url,
    untar=True,
    cache_dir='stack_overflow',
    cache_subdir=''
)
dataset_dir = pathlib.Path(dataset).parent

ผลลัพธ์

Downloading tensorflow-text
Downloading data from https://storage.googleapis.com/.../stack_overflow_16k.tar.gz
6053888/6053168 [==============================] - 0s 0us/step

คำอธิบาย

  • นำเข้าโมดูล Keras preprocessing, layers, losses

  • ใช้ TextVectorization layer สำหรับ NLP

  • ดาวน์โหลดข้อมูล 16k คำถาม StackOverflow