iT邦幫忙

2023 iThome 鐵人賽

DAY 23
0

以下是使用Python和常见的机器学习库(例如TensorFlow或PyTorch)来训练模型的典型代码示例,其中包括数据预处理步骤:
import numpy as np
import tensorflow as tf
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler

假设你有一个数据集X和对应的标签y

数据预处理步骤

1. 划分数据集为训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

2. 特征标准化(可选)

scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

创建模型

model = tf.keras.Sequential([
tf.keras.layers.Dense(64, activation='relu', input_shape=(X_train.shape[1],)),
tf.keras.layers.Dense(32, activation='relu'),
tf.keras.layers.Dense(1, activation='sigmoid')
])

编译模型

model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])

训练模型

model.fit(X_train, y_train, epochs=10, batch_size=32, validation_data=(X_test, y_test))

评估模型

test_loss, test_accuracy = model.evaluate(X_test, y_test)
print(f"Test Loss: {test_loss}, Test Accuracy: {test_accuracy}")

使用模型进行预测

predictions = model.predict(X_test)

by chatgpt


上一篇
超参数调优
下一篇
AI影片生成器,不用相機只要文字
系列文
不同的AI30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言