



'''simple docstring'''


import numpy as np
from matplotlib import pyplot as plt
from sklearn.datasets import load_iris
from sklearn.metrics import ConfusionMatrixDisplay
from sklearn.model_selection import train_test_split
from xgboost import XGBClassifier







def 				__lowerCamelCase				(   A__    )       ->	tuple:






   """simple docstring"""
   return (data["data"], data["target"])







def 				__lowerCamelCase				(   A__    ,						A__    )       ->	XGBClassifier:






   """simple docstring"""
   UpperCamelCase					=  XGBClassifier()
   classifier.fit(A__    ,						A__    )
   return classifier







def 				__lowerCamelCase				(   )       ->	None:






   """simple docstring"""
   UpperCamelCase					=  load_iris()
   UpperCamelCase				,      UpperCamelCase					=  data_handling(A__    )
   UpperCamelCase				,      UpperCamelCase				,      UpperCamelCase				,      UpperCamelCase					=  train_test_split(
       A__    ,						A__    ,						test_size=0.25    )

   UpperCamelCase					=  iris['target_names']

   # Create an XGBoost Classifier from the training data
   UpperCamelCase					=  xgboost(A__    ,						A__    )

   # Display the confusion matrix of the classifier with both training and test sets
   ConfusionMatrixDisplay.from_estimator(
       A__    ,						A__    ,						A__    ,						display_labels=A__    ,						cmap='Blues'    ,						normalize='true'    ,						)
   plt.title('Normalized Confusion Matrix - IRIS Dataset'    )
   plt.show()


if __name__ == "__main__":
   import doctest

   doctest.testmod(verbose=True)
   main()









import argparse
import os

import numpy as np
import tensorflow as tf
import torch

from transformers import BertModel






def   a			(						lowerCamelCase_		,   lowerCamelCase_		,   lowerCamelCase_				):






				'''simple docstring'''
				lowercase__    =			('''dense.weight''', '''attention.self.query''', '''attention.self.key''', '''attention.self.value''')

				lowercase__    =			(
				    ('''layer.''', '''layer_'''),
				    ('''word_embeddings.weight''', '''word_embeddings'''),
				    ('''position_embeddings.weight''', '''position_embeddings'''),
				    ('''token_type_embeddings.weight''', '''token_type_embeddings'''),
				    ('''.''', '''/'''),
				    ('''LayerNorm/weight''', '''LayerNorm/gamma'''),
				    ('''LayerNorm/bias''', '''LayerNorm/beta'''),
				    ('''weight''', '''kernel'''),
				)

				if not os.path.isdir(lowerCamelCase_				):
								os.makedirs(lowerCamelCase_				)

				lowercase__    =			model.state_dict()

				def to_tf_var_name(lowerCamelCase_				):
								for patt, repl in iter(lowerCamelCase_				):
												lowercase__    =			name.replace(lowerCamelCase_		,   lowerCamelCase_				)
								return F"""bert/{name}"""

				def create_tf_var(lowerCamelCase_		,   lowerCamelCase_		,   lowerCamelCase_				):
								lowercase__    =			tf.dtypes.as_dtype(tensor.dtype				)
								lowercase__    =			tf.get_variable(dtype=lowerCamelCase_		,   shape=tensor.shape		,   name=lowerCamelCase_		,   initializer=tf.zeros_initializer()				)
								session.run(tf.variables_initializer([tf_var]				)				)
								session.run(lowerCamelCase_				)
								return tf_var

				tf.reset_default_graph()
				with tf.Session() as session:
								for var_name in state_dict:
												lowercase__    =			to_tf_var_name(lowerCamelCase_				)
												lowercase__    =			state_dict[var_name].numpy()
												if any(x in var_name for x in tensors_to_transpose				):
																lowercase__    =			torch_tensor.T
												lowercase__    =			create_tf_var(tensor=lowerCamelCase_		,   name=lowerCamelCase_		,   session=lowerCamelCase_				)
												tf.keras.backend.set_value(lowerCamelCase_		,   lowerCamelCase_				)
												lowercase__    =			session.run(lowerCamelCase_				)
												print(F"""Successfully created {tf_name}: {np.allclose(lowerCamelCase_		,   lowerCamelCase_				)}"""				)

								lowercase__    =			tf.train.Saver(tf.trainable_variables()				)
								saver.save(lowerCamelCase_		,   os.path.join(lowerCamelCase_		,   model_name.replace('''-'''		,   '''_'''				) + '''.ckpt'''				)				)






def   a			(						lowerCamelCase_=None				):






				'''simple docstring'''
				lowercase__    =			argparse.ArgumentParser()
				parser.add_argument('''--model_name'''		,   type=lowerCamelCase_		,   required=lowerCamelCase_		,   help='''model name e.g. bert-base-uncased'''				)
				parser.add_argument(
				    '''--cache_dir'''		,   type=lowerCamelCase_		,   default=lowerCamelCase_		,   required=lowerCamelCase_		,   help='''Directory containing pytorch model'''				)
				parser.add_argument('''--pytorch_model_path'''		,   type=lowerCamelCase_		,   required=lowerCamelCase_		,   help='''/path/to/<pytorch-model-name>.bin'''				)
				parser.add_argument('''--tf_cache_dir'''		,   type=lowerCamelCase_		,   required=lowerCamelCase_		,   help='''Directory in which to save tensorflow model'''				)
				lowercase__    =			parser.parse_args(lowerCamelCase_				)

				lowercase__    =			BertModel.from_pretrained(
				    pretrained_model_name_or_path=args.model_name		,   state_dict=torch.load(args.pytorch_model_path				)		,   cache_dir=args.cache_dir		,   )

				convert_pytorch_checkpoint_to_tf(model=lowerCamelCase_		,   ckpt_dir=args.tf_cache_dir		,   model_name=args.model_name				)


if __name__ == "__main__":
							main()




