model_quantum.py

def make_model_quantum(n_categories, n_qubits=4, n_layers=2, embedding_dim=512):

    text_in = keras.Input( shape=(embedding_dim,), dtype=tf.float64, name='text_in')

    x = layers.Dense(n_qubits, activation='tanh', dtype=tf.float64)(text_in)

    x = VariationalQuantumCircuit(
            n_categories=n_categories, 
            n_qubits=n_qubits, 
            n_layers=n_layers)(x)
    
    assert x.shape[-1] == n_qubits

    x_out = layers.Dense(n_categories, activation='softmax', dtype=tf.float64)(x)

    return keras.Model(inputs=text_in, outputs=x_out, name="QuantumPreprintClassifier")