機械学習アルゴリズムに2つのクラス分類問題を与えます。入力データを受け付ける必要がありますが、どのクラスに属するかを決定するアルゴリズムを書いてください。
def classify_data(X):
    # Define model
    model = Sequential()
    model.add(Dense(1, activation="sigmoid"))
    model.compile(loss="binary_crossentropy", optimizer="adam", metrics=["accuracy"])
    
    # Fit model 
    model.fit(X, y, batch_size=10, epochs=10, verbose=True)
    
    # Predict class
    pred_class = model.predict(X)
    
    return pred_class