Hello! I am trying to train a neural network model and convert using tflite (TensorFlow), but my archive.cc generated there is errors when I compile the code main.cpp.
“Null(s) character(s) ignored [Ln,Col]”
“Stray ‘\377’ in program”
The code is:
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
from tensorflow.keras import Input
from tensorflow.data import Dataset
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import OneHotEncoder, MinMaxScaler, StandardScaler
from sklearn import datasets
import numpy as np
import os
import itertools
data = datasets.load_iris()
X = data['data']
Y = data['target']
# use and the scaled input the encoded output
enc = OneHotEncoder()
Y_enc = enc.fit_transform(Y[:, np.newaxis]).toarray()
Y = Y_enc
scaler = MinMaxScaler()
X_scaled = scaler.fit_transform(X)
X = X_scaled
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.3, random_state=0)
num_features = X.shape[1]
num_classes = Y.shape[1]
batch_size = 1
epochs = 100
model = Sequential([
Input(shape=(4)),
Dense(5, activation='relu'),
Dense(3, activation='softmax')
])
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
history = model.fit(X_train, Y_train,
batch_size=batch_size,
epochs=epochs,
verbose=1,
)
score_train = model.evaluate(X_train, Y_train, verbose=0)
print('Train loss:', score_train[0])
print('Train accuracy:', score_train[1])
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_quant_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_quant_model)