cc_train_chain.py

import numpy as np
import scipy.sparse 
import tensorflow as tf
import tensorflow.keras as keras
from classifier_chains import ClassifierChain

np.random.seed(0)
tf.set_random_seed(0)

X_train = scipy.sparse.load_npz('X_train.npz')
X_test = scipy.sparse.load_npz('X_test.npz')

y_train = np.load('y_train.npy')

model = keras.models.Sequential()
model.add(keras.layers.Dense(units=50, input_dim=X_train.shape[1], activation='relu'))
model.add(keras.layers.Dropout(0.1))
model.add(keras.layers.Dense(units=2, input_dim=50, activation='softmax'))

sgd_optimizer = keras.optimizers.SGD(lr=0.1, decay=1e-15, momentum=.9)

model.compile(optimizer=sgd_optimizer, loss='binary_crossentropy')

optimizers= [sgd_optimizer]
losses = ['binary_crossentropy']

cc = ClassifierChain(classifier=model, num_labels=2, name='AG_NT_chain', optimizers=optimizers, losses=losses, create_missing=True)

debalancing = [0, 1.4] 
cc.fit(X=X_train, y=y_train, epochs=15, batch_size=64, weights_mode='chain', debalancing=debalancing)

preds_train_AG_NT_chain = cc.predict(X_train)

preds_test_AG_NT_chain = cc.predict(X_test)

np.save('preds_train_AG_NT_chain.npy', preds_train_AG_NT_chain)

np.save('preds_test_AG_NT_chain', preds_test_AG_NT_chain)