U f@sddlZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZddlmZddlZddlZddlZdejd<d d Zd d Zd dZddZdS)N) int_shape) Input Cropping1DaddConv1DGlobalAvgPool1DDenseAdd ConcatenateLambdaFlatten)Adam)Model)multinomial_nll0ZPYTHONHASHSEEDcCs^ddlm}ddlm}ttd}||||}t|j}t |D]}d|j|_ qH|S)Nr) load_model)get_custom_objects)rtfF) tensorflow.keras.modelsrZtensorflow.keras.utilsrrrupdatelenZlayersrangeZ trainable)Z model_hdf5rrZcustom_objectsZpretrained_bias_modelZ num_layersirK/tmp/pip-target-lpfmz8o1/lib/python/../../bin/chrombpnet_with_bias_model.pyload_pretrained_biass      rc Csd}d}d}t|dfdd}t||ddd d |}d d td|dD} td|dD]} d | | d} t|dddd| | d|} t|d} t| d}| |ddkstt| |dd| | dd|}t| |g}qXt||ddd|}tt|ddt|d}|dks,tt|dddksFtt|dd|}t dd|}t dd|}t |dd|}t |g||gdd}|S)NKsequenceshapenamevalidZreluZwo_bias_bpnet_1st_conv) kernel_sizepadding activationr#cSsg|] }t|qSr)str).0rrrr +szbpnet_model..zwo_bias_bpnet_{}conv)r%r&r'Z dilation_rater#rzwo_bias_bpnet_{}cropr#Zwo_bias_bpnet_prof_out_precrop)filtersr%r&r#Z#wo_bias_bpnet_logitt_before_flattenZ(wo_bias_bpnet_logits_profile_predictionsgapZ"wo_bias_bpnet_logcount_predictions model_wo_bias)inputsoutputsr#) rrrformatrAssertionErrorrrintr rrr)r. n_dil_layers sequence_len out_pred_lenZconv1_kernel_sizeZprofile_kernel_sizeZ num_tasksinpxZ layer_namesrZconv_layer_nameZconv_xZx_lenZ conv_x_lenZprof_out_precropZcropsizeZprof profile_outZgap_combined_conv count_outmodelrrr bpnet_modelsb  & r>cCsd|kstt|d}t|d}t|d}|d}t|d}t|d}t|}t||||} |j} tj| t j | t | t |dfdd } || } | | } t | d jd kstt | d jd kstt | d jd kstt | d jd ks t| d jd d ks$t| d jd |ksz9getModelGivenModelOptionsAndWeightInits..Zlogcount_predictionsr1r2) learning_rateZmse)Z optimizerZlossZ loss_weights)keysr4r5floatrr>seednprandomrZset_seedrnrrr"r r r rcompiler rGr)argsZ model_paramsr.r6r@r?r7r8Z bias_modelZbpnet_model_wo_biasrJr9Z bias_outputZoutput_wo_biasr;Z concat_countsr<r=rrr'getModelGivenModelOptionsAndWeightInits\sF        rPcCsF|dj}t|dj|d|dgd}td||ddS)Nr0rrrFzsave model without biasz _nobias.h5)Z get_layeroutputrr1printsave)r=Z output_prefixr0Zmodel_without_biasrrrsave_model_without_biass  rT)numpyrKZtensorflow.keras.backendrZtensorflow.keras.layersrrrrrrr r r r Ztensorflow.keras.optimizersr rrZ chrombpnet.training.utils.lossesrZ tensorflowrrLrMosenvironrr>rPrTrrrrs 0     A1