Jonas Oppenlaender joetm

🎯

Focusing

joetm / genSankey

Created May 22, 2023 10:04 — forked from ken333135/genSankey

Wrapper Function to create Sankey Diagram from DataFrame

	def genSankey(df,cat_cols=[],value_cols='',title='Sankey Diagram'):
	# maximum of 6 value cols -> 6 colors
	colorPalette = ['#4B8BBE','#306998','#FFE873','#FFD43B','#646464']
	labelList = []
	colorNumList = []
	for catCol in cat_cols:
	labelListTemp = list(set(df[catCol].values))
	colorNumList.append(len(labelListTemp))
	labelList = labelList + labelListTemp

joetm / count_ckpt_param.py

Created July 7, 2022 08:46 — forked from dalgu90/count_ckpt_param.py

Count the number of parameter in a TensorFlow checkpoint file. (Usage: python count_ckpt_param.py path-to-ckpt)

	#!/usr/bin/env python

	import sys
	import tensorflow as tf
	import numpy as np

	if len(sys.argv) == 2:
	ckpt_fpath = sys.argv[1]
	else:
	print('Usage: python count_ckpt_param.py path-to-ckpt')